记录pytorch的shape是怎么变化的,主要是unsquezze函数
代码如下:
a = torch.tensor([[[1,2,3,4],[0,0,0,0],[7,8,9,10]],[[1,4,9,16],[16,25,36,49],[49,64,81,100]]],dtype=torch.float)
len_q = a.size(1)
padding_mask = a.eq(0)
print('1,',padding_mask)
print('2,',padding_mask[:,:,0])
pr