import torch
torch.squeeze(input)
一、squeeze
的作用是什么?
压缩张量维度,即当张量的某个维度的成员数量为1时,直接删除该维度。(注意,张量的维度由外层向内层看,分别是0维,1维,2维···)
二、对squeeze
的作用的直观理解
三、应用
代码一:
#以一个三维张量为例
t1 = torch.tensor([[[0,1,2],[3,4,5]]]) #torch.size([1,2,3])
t2 = torch.tensor([[