So, I found this cool Normalizing flow tutorial in PyTorch and I was trying the first tut itself link here
import torch.distributions as distrib import torch.distributions.transforms as transforms x = np.linspace(-4, 4, 1000) z = np.array(np.meshgrid(x, x)).transpose(1, 2, 0) z = np.reshape(z, [z.shape * z.shape, -1])
And the transformations like
# Initial distribution q0 = distrib.MultivariateNormal(torch.zeros(2), torch.eye(2)) # Defining Affine Transformation f1 = transforms.ExpTransform() # Transforming q1 = distrib.TransformedDistribution(q0, f1) ---> q1.log_prob(torch.Tensor(z))
The last line raises
ValueError: The value argument must be within the support which is not the case in the notebook did something change? Because I tried running the whole notebook on colab and it broke there too and on my system too.
Any help what might be wrong?