博客

torch.randn()

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

返回一个符合均值为0，方差为1的正态分布（标准正态分布）中填充随机数的张量


Parameters


• 
  size(int…)
  
  –定义输出张量形状的整数序列。可以是数量可变的参数，也可以是列表或元组之类的集合。


Keyword Arguments


• 
  
out

  (Tensor, optional)
  
  –输出张量
• 
  
dtype

  (torch.dtype, optional)
  
  –返回张量所需的数据类型。默认:如果没有，使用全局默认值
• 
  
layout

  (torch.layout, optional)
  
  –返回张量的期望布局。默认值:
  
torch.strided

• 
  
device

  (torch.device, optional)
  
  –返回张量的所需 device。默认:如果没有，则使用当前设备作为默认张量类型.(CPU或CUDA)
• 
  
requires_grad

  (bool, optional)
  
  –
  
autograd

  是否应该记录对返回张量的操作(说明当前量是否需要在计算中保留对应的梯度信息)。默认值:
  
False

  。

Example:

>>> torch.randn(4)
tensor([-2.1436,  0.9966,  2.3426, -0.6366])
>>> torch.randn(2, 3)
tensor([[ 1.5954,  2.8929, -1.0923],
        [ 1.1719, -0.4709, -0.1996]])