pytorch tensor的通道排序

pytorch tensor的通道排序 [batch, channel, height, width]

batch:batch_size

channel:如彩色图像RGB,channel=3

height:图片的高度

width:图片的宽度

你可能感兴趣的