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([[ 1.7141, -0.2354],
        [-0.4287, -0.0797],
        [-1.3418, -2.5155],
        ...,
        [-0.5308,  1.3941],
        [-0.9714, -1.0535],
        [-0.0363, -1.4620]])