维度变换view.reshape:torch(4,1,28,28)#4张1维(channel)的灰 ... b.expand(-1,32,-1,-1) #-1表示该dimension不扩充,只扩充第二个dimension.
確定! 回上一頁