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

Question

我有一个我使用创建的张量

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

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

如何添加泊松噪声和高斯噪声？

但它似乎与图像有关。

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)