python torch randn
时间: 2024-01-16 12:15:05 浏览: 129
Python库 | torch_sparse-0.2.0.tar.gz
torch.randn是一个用来生成随机数字的函数,这些随机数字满足标准正态分布(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]])
阅读全文