本篇文章给大家分享的是有关torch.Tensor.size()方法如何使用,小编觉得挺实用的,因此分享给大家学习,希望大家阅读完这篇文章后可以有所收获,话不多说,跟着小编一起来看看吧。
该方法返回的是当前张量的形状,返回值是元组tuple的一个子类.
代码实验举例:
(base) PS C:\Users\chenxuqi> python Python 3.7.4 (default, Aug 9 2019, 18:34:13) [MSC v.1915 64 bit (AMD64)] :: Anaconda, Inc. on win32 Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>> a = torch.ones(2,3,4)>>> a tensor([[[1., 1., 1., 1.], [1., 1., 1., 1.], [1., 1., 1., 1.]],[[1., 1., 1., 1.], [1., 1., 1., 1.], [1., 1., 1., 1.]]])>>> a.size<built-in method size of Tensor object at 0x000001D0D5CAE368>>>>>>> a.size()torch.Size([2, 3, 4])>>>>>> a.size(0)2>>> a.size()[0]2>>>>>>>>> a = torch.ones(2,3,4,5,6,7,8,9)>>> a.shape torch.Size([2, 3, 4, 5, 6, 7, 8, 9])>>> a.size()torch.Size([2, 3, 4, 5, 6, 7, 8, 9])>>> a.size(0)2>>> a.size(7)9>>> a.size()[0]2>>> a.size()[7]9>>> a.size(4)6>>> a.size()[4]6>>>>>>>>> a.size()[:]torch.Size([2, 3, 4, 5, 6, 7, 8, 9])>>>>>> a.size()[4:]torch.Size([6, 7, 8, 9])>>>>>>>>>>>>
以上就是torch.Tensor.size()方法如何使用,小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注亿速云行业资讯频道。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。