pytorch中的unsqueeze以及squeeze用法舉例

unsqueeze：簡單來說就是添加tensor的維度 舉例說明： 代碼 import torch x = torch.tensor([1, 2, 3])#dim=1,即（3） print('x: ', x) print('x.size: ', x.size()) x1 = torch.unsqueeze(x, 1)#x1變爲（3,1）的矩陣 print('x1: ', x1) print('x1

>>阅读原文<<