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
·
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]])
更多推荐
已为社区贡献11条内容
所有评论(0)