关于reshape和交换通道的问题
来源:3-4 卷积神经网络实战
慕标0697996
2019-05-18
老师您好,同学们好,我不知道为什么x需要先reshape再交换通道,不能直接写成[-1,32,32,3]吗?是不是这个x就是这么规定的第二位是张量,第三位和第四位是图片的大小(32*32)?之后又为什么要交换通道?
写回答
1回答
-
这是由我们用的数据的存储格式决定的,在数据中,每个样本长度是3072,前1024,中间1024和最后1024分别对应一个通道,所以reshape后必须是[3, 32, 32]的。当然,如果数据存储的是其他格式,那么我们reshape和交换通道的手段也要不一样,你可以自由定义你存储的数据的格式。但在课程中,我们用的数据是别人定义的,所以我们要遵循他们的规则。但要注意,这不是固定的。
112019-05-28
相似问题