PyTorch:全局函数

Posted -柚子皮-

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PyTorch:全局函数相关的知识,希望对你有一定的参考价值。

torch.randn

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

outi​∼N(0,1)

示例

>>> 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]])

# 生成一个4*3*2维的张量

input = torch.randn(4, 3, 2)

from: -柚子皮-

ref:

以上是关于PyTorch:全局函数的主要内容,如果未能解决你的问题,请参考以下文章

PyTorch:tensor-数据处理

PyTorch:tensor-数据处理

PyTorch:模型训练和预测

PyTorch:模型训练和预测

PyTorch:全局函数

PyTorch:全局函数