I would like to create a batch of identity matrices to initialize a distributions.MultivariateNormal object. Is there a way to call torch.eye() as a batch operation?
If not, are there any alternatives you would suggest?