tensor 添加维度

tensor 添加维度

	data = images.unsqueeze(1)

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

你可能感兴趣的:(pytorch,python,开发语言)