https://github.com/pytorch/vision/pull/8381#discussion_r1620941790 https://pytorch.org/docs/stable/distributions.html#torch.distributions.normal.Normal
