torch.randn(20, 5) 的用法
时间: 2023-07-21 17:09:27 浏览: 80
pytorch中torch.max和Tensor.view函数用法详解
5星 · 资源好评率100%
torch.randn(20, 5) 是 PyTorch 中用于生成一个大小为 (20, 5) 的张量,其中的元素是从标准正态分布(均值为0,方差为1)中随机采样得到的。
具体来说,torch.randn() 函数可以接受任意数量的整数作为输入,这些整数指定了生成张量的形状。在本例中,我们传入了两个整数 20 和 5,因此生成了一个 20 行 5 列的张量。这个张量的值是随机生成的,但它们的分布遵循标准正态分布。
需要注意的是,torch.randn() 函数生成的张量是不可导的(即不能用于反向传播),如果需要可导的随机张量,可以使用 torch.randn_like() 或 torch.rand() 函数。
阅读全文