删除tensor中维度为1的维度在pytorch中,用torch.squeeze()函数 ... e = torch.unsqueeze(x, dim=0) # 在第一维度添加维度 print(e.shape) f ...
確定! 回上一頁