torch.randn()是一个可以生成随机数的函数,返回一个张量,张量中的值是从标准正态分布中随机抽取的。它可以用来初始化神经网络中的权重或者偏置等参数。该函数的语法格式如下:

torch.randn(*size, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

其中:

  • size(int...):生成的随机数张量的形状。
  • out(Tensor, optional):输出张量。
  • dtype(torch.dtype, optional):输出张量的数据类型。默认是float32。
  • layout(torch.layout, optional):输出张量的布局。默认是torch.strided。
  • device(torch.device, optional):输出张量的设备。默认是使用当前设备。
  • requires_grad(bool, optional):输出张量是否需要梯度。默认为False。

举个例子,生成一个形状为(3, 4)的张量:

import torch

x = torch.randn(3, 4)
print(x)

输出结果:

tensor([[ 1.1313,  0.4642,  0.5848, -0.2009],
        [-1.2127, -0.7511, -0.1676, -0.5092],
        [ 0.9421,  0.1234, -2.1395,  0.5100]])

可以看到,生成了一个形状为(3, 4)的张量,张量中的值是从标准正态分布中随机抽取的。

介绍一下torchrandn并且举例

原文地址: https://www.cveoy.top/t/topic/Nr1 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录