torch.normal函数

torch.normal是PyTorch中的一个函数，用于生成服从正态分布的随机数。它的语法如下：

torch.normal(mean, std, out=None)

其中，mean是一个张量，表示正态分布的均值；std是一个张量，表示正态分布的标准差；out是一个可选参数，表示输出的张量。如果不指定out，则函数会返回一个新的张量。

例如，下面的代码生成一个均值为0，标准差为1的正态分布随机数张量：

import torch

x = torch.normal(mean=torch.zeros(3, 3), std=torch.ones(3, 3))
print(x)

输出：

tensor([[ 0.0239, -0.6317,  1.2278],
        [-0.0777, -0.3716,  1.0573],
        [-0.9482, -1.1047, -0.4047]])

这个张量中的每个元素都是从均值为0，标准差为1的正态分布中采样得到的。