torch.randn 是一个 PyTorch 中的函数,用于生成指定大小的张量,其中每个元素都是从标准正态分布(均值为0,标准差为1)中随机抽取的。
torch.randn 的语法如下:
torch.randn(*size, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor
其中 *size 表示张量的大小,可以是一个整数,一个元组或多个整数。例如,要创建一个大小为 3x2 的张量,可以使用以下代码:
import torch
x = torch.randn(3, 2)
print(x)
输出结果:
tensor([[ 0.4438, -0.0241],
[-0.4326, -0.8158],
[-0.3517, -1.3522]])
在上面的代码中,我们创建了一个大小为 3x2 的张量 x,其中每个元素都是从标准正态分布中随机抽取的。
out 参数可以指定一个输出张量,将生成的随机数填充到这个张量中。
dtype 参数可以指定生成的随机数的数据类型。默认情况下,它是 torch.float32。
device 参数可以指定生成的张量的计算设备,例如 CPU 或 GPU。文章来源:https://www.toymoban.com/news/detail-404487.html
requires_grad 参数可以指定是否需要计算梯度,默认值为 False。文章来源地址https://www.toymoban.com/news/detail-404487.html
到了这里,关于torch.randn的用法的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!