torch.normal() 的用法
该函数的参数如下:
normal(mean, std, *, generator=None, out=None)
参数说明
-
mean
:- 均值,可以是一个数值(标量)或者张量。
- 如果是张量,则指定生成正态分布的均值,形状需与标准差匹配。
-
std
:- 标准差,可以是一个数值(标量)或者张量。
- 如果是张量,则指定生成正态分布的标准差,形状需与均值匹配。
-
generator
(可选): 用于生成随机数的随机数生成器。 -
out
(可选): 如果提供,则结果存储在这个张量中。
返回值
返回一个与 mean
和 std
的形状匹配的张量,值是符合指定正态分布的随机数。
函数使用示例
生成一个标量值
import torch
# 生成均值为0,标准差为1的正态分布随机数
result = torch.normal(0, 1)
print(result)
<