评价此页

torch.randn_like#

torch.randn_like(input, *, dtype=None, layout=None, device=None, requires_grad=False, memory_format=torch.preserve_format) Tensor#

返回一个与 input 大小相同的张量,其中填充了来自均值为 0、方差为 1 的正态分布的随机数。有关复数数据类型(complex dtypes)的采样过程,请参阅 torch.randn()torch.randn_like(input) 等效于 torch.randn(input.size(), dtype=input.dtype, layout=input.layout, device=input.device)

参数

input (Tensor) – input 的大小将决定输出张量的大小。

关键字参数
  • dtype (torch.dtype, optional) – 返回张量所需的数据类型。默认值:如果为 None,则默认为 input 的数据类型。

  • layout (torch.layout, optional) – 返回张量所需的布局。默认值:如果为 None,则默认为 input 的布局。

  • device (torch.device, optional) – 返回张量所需的设备。默认值:如果为 None,则默认为 input 的设备。

  • requires_grad (bool, optional) – 如果自动求导(autograd)应该记录返回张量上的操作。默认值:False

  • memory_format (torch.memory_format, optional) – 返回张量所需的内存格式。默认值:torch.preserve_format