torch.randn
时间: 2023-07-11 07:55:50 浏览: 71
pyTorchRNN
torch.randn 是 PyTorch 中的一个函数,它可以生成指定形状的张量,并且张量中的元素都是从标准正态分布中随机抽取的。具体来说,它的用法是 torch.randn(*sizes, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False),其中 sizes 是一个整数元组,表示生成的张量的形状;out 是一个可选的输出张量;dtype 是生成张量的数据类型;layout 是生成张量的布局;device 是生成张量的设备;requires_grad 表示生成的张量是否需要计算梯度。
阅读全文