torch.randn
时间: 2023-10-31 11:49:43 浏览: 4
torch.randn是PyTorch中的一个函数,用于生成指定形状的随机数张量,张量中的元素服从标准正态分布(均值为0,标准差为1)。该函数的语法格式为:torch.randn(*size, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False),其中,*size表示张量的形状,out为输出张量(可选),dtype为数据类型(可选),layout表示张量的布局方式(可选),device表示张量存储的设备(可选),requires_grad表示是否开启梯度跟踪(可选)。
阅读全文