torch.normal

torch.normal#

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

返回一个张量,包含从给定参数means,std的离散正态分布中抽取随机数。 均值means是一个张量,包含每个输出元素相关的正态分布的均值。 std是一个张量,包含每个输出元素相关的正态分布的标准差。 均值和标准差的形状不须匹配,但每个张量的元素个数须相同。

参数:

import torch    
w = torch.tensor([2, -3.4])
X = torch.normal(0, 1, (1000, len(w)))
X
tensor([[ 0.6342,  1.1755],
        [-0.6606, -0.7874],
        [-0.3401, -0.5856],
        ...,
        [-0.4464, -0.1336],
        [-0.5167,  1.0033],
        [-1.0096, -0.4360]])