torch.normal的作用
时间: 2024-05-24 10:11:28 浏览: 143
Pytorch中.new()的作用详解
torch.normal用于生成一个张量,其中包含从指定均值和标准差的正态分布中提取的随机数。它的作用类似于numpy.random.normal函数。
函数签名如下:
torch.normal(mean, std, out=None) → Tensor
其中:
- mean: 一个张量,包含每个输出元素所需的正态分布的均值。
- std: 一个张量,包含每个输出元素所需的正态分布的标准差。
- out: 可选的输出张量。
示例代码:
import torch
# 生成均值为0,标准差为1的正态分布随机数
x = torch.normal(mean=0, std=1, size=(3,3))
print(x)
输出结果:
tensor([[ 0.7763, 0.9473, -0.1142],
[-0.6795, 0.2097, 0.4959],
[-2.2547, -0.3830, -1.3248]])
阅读全文