torch.Tensor.size()方法如何使用

本篇文章给大家分享的是有关torch.Tensor.size()方法如何使用,小编觉得挺实用的,因此分享给大家学习,希望大家阅读完这篇文章后可以有所收获,话不多说,跟着小编一起来看看吧。

公司主营业务:网站设计制作、网站建设、移动网站开发等业务。帮助企业客户真正实现互联网宣传,提高企业的竞争能力。创新互联是一支青春激扬、勤奋敬业、活力青春激扬、勤奋敬业、活力澎湃、和谐高效的团队。公司秉承以“开放、自由、严谨、自律”为核心的企业文化,感谢他们对我们的高要求,感谢他们从不同领域给我们带来的挑战,让我们激情的团队有机会用头脑与智慧不断的给客户带来惊喜。创新互联推出肃南裕固族自治免费做网站回馈大家。

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>>>>>> 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()方法如何使用,小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注创新互联行业资讯频道。


分享标题:torch.Tensor.size()方法如何使用
URL地址:http://scjbc.cn/article/jhedps.html

其他资讯