torch.normal
时间: 2023-10-05 07:14:01 浏览: 60
Pytorch生成随机数Tensor的方法汇总
`torch.normal(mean, std, out=None)`函数是PyTorch中的一个函数,用于产生服从正态分布的随机数。其中,`mean`是一个张量,表示正态分布的均值;`std`是一个张量,表示正态分布的标准差;`out`是一个可选参数,表示输出张量。
该函数的输出是一个张量,其形状与`mean`和`std`相同,其中每个元素都是从对应的正态分布中抽取的随机数。
例如,下面的代码使用`torch.normal`函数产生一个形状为`(3, 2)`的张量,其中每个元素都是从均值为0、标准差为1的正态分布中抽取的随机数:
```python
import torch
mean = torch.zeros(3, 2)
std = torch.ones(3, 2)
out = torch.normal(mean=mean, std=std)
print(out)
```
输出结果为:
```
tensor([[-0.1524, -0.4292],
[-0.1109, 0.2393],
[-1.2703, 0.0114]])
```
可以看到,输出的张量中的每个元素都是一个随机数,其值符合正态分布。
阅读全文