tensor 添加维度
data = images.unsqueeze(1)
squeeze() 压缩 <–> unsqueeze() 扩展
参数 index -> 1 表示在第1维度之后插入
0 表示在第0维度之后插入,即最前面插入
tensor 添加维度
data = images.unsqueeze(1)
squeeze() 压缩 <–> unsqueeze() 扩展
参数 index -> 1 表示在第1维度之后插入
0 表示在第0维度之后插入,即最前面插入