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

Question

我有一个使用创建的张量

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

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

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

但它似乎与图像有关。

标签： pytorch

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)