概述
作用：查看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.size()函数所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复