torch.randn用法

torch.randn:用来生成随机数字的tensor，这些随机数字满足标准正态分布（0~1）。torch.randn（size),size可以是一个整数，也可以是一个元组。代码示例：import torcha=torch.randn(3)b=torch.randn(3,4)print("a:",a)print("b:",b)输出：a: tensor([ 0.9405, -0.1068,0.171

江南汪

77274人浏览 · 2021-05-10 21:42:16

江南汪 · 2021-05-10 21:42:16 发布

torch.randn:用来生成随机数字的tensor，这些随机数字满足标准正态分布（0~1）。
torch.randn（size),size可以是一个整数，也可以是一个元组。
代码示例：

import torch
a=torch.randn(3)
b=torch.randn(3,4)
print("a:",a)
print("b:",b)

输出：

a: tensor([ 0.9405, -0.1068,  0.1712])
b: tensor([[-1.0962, -0.1893,  1.2323,  0.5703],
        [-1.5256, -1.4923,  0.4275,  0.5143],
        [ 1.1200,  0.5317,  1.1961, -2.2533]])