Searched defs:event_shape (Results 1 – 4 of 4) sorted by relevance
326 def test_transformed_distribution_shapes(batch_shape, event_shape, dist): argument
109 def event_shape(self) -> torch.Size: member in Distribution
239 event_shape=None, argument
690 def event_shape(self): member in Distribution