pytorch view详解

介绍:view是pytorch中tensor的一种方法

tensor.view(shape)

功能:将tensor变成另外另外一个大小与shape相同的tensor

实例:

>>> x = torch.randn(4, 4)
>>> x.size()
torch.Size([4, 4])
>>> y = x.view(16)
>>> y.size()
torch.Size([16])

特别的当shape中有某个维数为-1时,则该维的维数会根据其他维的维数自动确定

实例:

>>> z = x.view(-1, 8) 
>>> z.size()
torch.Size([2, 8])      #2*8=16 所以-1的维数为2
发布了8 篇原创文章 · 获赞 5 · 访问量 1411

猜你喜欢

转载自blog.csdn.net/Li7819559/article/details/103813117