torch.normal函数的用法和注意事项

torch.normal函数用于从指定均值和标准差的正态分布中生成随机数。它的用法如下：

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

参数说明：
- mean：一个张量或浮点数，表示正态分布的均值。
- std：一个张量或浮点数，表示正态分布的标准差。
- out（可选）：输出张量。

注意事项：
- mean和std可以是标量（单个值）或张量（与输出张量大小相同）。
- 如果mean和std都是标量，则返回一个具有形状为out的张量。
- 如果mean和std都是张量，则它们的形状必须广播到相同形状，并返回一个具有相同形状的张量。

示例代码：

import torch

mean = 0
std = 1
out = torch.empty(3, 3)

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