Torch.distributions with cuda tensor input?

1

Is it possible to use cuda tensor as an input in torch.distributions?
For example, everytime I use torch.distributions.normal.Normal, Normal.cdf(), and pdf() do not accept cuda tensor.