I would like to calculate the log probability of a vector with respect to n different Gaussians, shown below.
from torch import tensor from pyro import distributions as dist data = tensor([0., 1., 10., 11., 12.]) dist.Normal(tensor([0., 10.]), tensor(1.)).log_prob(data) ValueError: Value is not broadcastable with batch_shape+event_shape: torch.Size() vs torch.Size().
Ideally, this would return a 5x2 vector of log probabilities. Is there a way to accomplish this?