pytorch使用过程中,经常需要对张量进行维度校准。我们简单把一个2x2的tensor升维到1x2x2,该怎么做呢?
方法一:a[None]
import torch
a = torch.rand(2, 2)
b = a[None]
print(b.shape)
# torch.Size([1, 2, 2])
方法二:a.unsqueeze(0)
import torch
a = torch.rand(2, 2)
b = a.unsqueeze(0)
print(b.shape)
# torch.Size([1, 2, 2])
虽然方法二比方法一更繁琐,但比方法一更加灵活。如果你想升维度2x2->2x1x2,可以:文章来源:https://www.toymoban.com/news/detail-521839.html
b = a.unsqueeze(1)
print(b.shape)
# torch.Size([2, 1, 2])
同理,a.unsqueeze(2)
也可以升维成(2,2,1)。方法一的优势是,看起来更老练。文章来源地址https://www.toymoban.com/news/detail-521839.html
到了这里,关于Pytorch张量升维的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!