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

>>阅读原文<<