Hello everyone,
is there a general way to reshape the batch_shape of a torch.distributions.Distribution (Probability distributions - torch.distributions — PyTorch 1.8.1 documentation)?
Similar to expand (Probability distributions - torch.distributions — PyTorch 1.8.1 documentation) I would expect a reshape method, but I cannot find one.
Best,
Tim