PyTorch | torch.zeros()怎么使用？ torch.view(-1)如何使用？
Posted 人工智能博士
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了PyTorch | torch.zeros()怎么使用？ torch.view(-1)如何使用？相关的知识，希望对你有一定的参考价值。
我不喜欢讲解，就喜欢直观，所以就看例子就行！
torch.zeros(2,3,4)
#输出：
tensor([[[0., 0., 0., 0.],
[0., 0., 0., 0.],
[0., 0., 0., 0.]],
[[0., 0., 0., 0.],
[0., 0., 0., 0.],
[0., 0., 0., 0.]]])view是自动计算匹配的！
import torch
a = torch.arange(0,20) #此时a的shape是(1,20)
a.view(4,5).shape #输出为(4,5)
a.view(-1,5).shape #输出为(4,5)
a.view(4,-1).shape #输出为(4,5)
a.view(10,-1).shape #输出为(10,2)
以上是关于PyTorch | torch.zeros()怎么使用？ torch.view(-1)如何使用？的主要内容，如果未能解决你的问题，请参考以下文章