torch.randn函数,torch语言

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.strideddevice(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]])