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