如何在PyTorch中向张量添加一些高斯噪声？

Question

我有一个使用创建的张量

    temp = torch.zeros(5, 10, 20, dtype=torch.float64)
    ## some values I set in temp

现在，我想向每个temp [i，j，k]添加一个高斯噪声（从均值0和方差0.1的正态分布中采样）。我该怎么做？我希望有一个函数可以张量噪声，但找不到任何东西。我确实找到了：

How to add Poisson noise and Gaussian noise?

但它似乎与图像有关。

Answer 1

函数torch.randn产生一个张量，该张量的元素均来自零均值和单位方差的高斯分布。乘以sqrt(0.1)即可得到所需的方差。

x = torch.zeros(5, 10, 20, dtype=torch.float64)
x = x + (0.1**0.5)*torch.randn(5, 10, 20)