作用：查看Tensor的维度。
具体例子如下：
>>> import torch
>>> a = torch.Tensor([[[1,1],[2,2]],[[3,3],[4,4]]])
>>> a.size()
torch.Size([2, 2, 2])
>>> a = torch.Tensor([[[1,1,1],[2,2,2]],[[3,3,3],[4,4,4]]])
>>> a.size()
torch.Size([2, 2, 3])
>>> a.size(0)
2
>>> a.size(1)
2
>>> a.size(2)
3
>>> a.size(-1)
3
最后
以上就是无心曲奇最近收集整理的关于深度学习框架_PyTorch_torch.size()函数的全部内容，更多相关深度学习框架_PyTorch_torch内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复