I have a normal distribution with means and variances whose shape are [32, 1, 28, 28].
I want to get ten samples from the normal distribution as follows.
x = mean + variance * epsilon (epsilon is sampled from N(0,1) )
First, I call torch.normal(0, 1, size=[32, 1, 28, 28, 10]).
And then I want to product variance and epsilon, resulting in a [32, 1, 28, 28, 10] matrix.
How can I do it?