关于reshape和交换通道的问题

来源:3-4 卷积神经网络实战

慕标0697996

2019-05-18

老师您好,同学们好,我不知道为什么x需要先reshape再交换通道,不能直接写成[-1,32,32,3]吗?是不是这个x就是这么规定的第二位是张量,第三位和第四位是图片的大小(32*32)?之后又为什么要交换通道?

写回答

1回答

正十七

2019-05-27

这是由我们用的数据的存储格式决定的,在数据中,每个样本长度是3072,前1024,中间1024和最后1024分别对应一个通道,所以reshape后必须是[3, 32, 32]的。当然,如果数据存储的是其他格式,那么我们reshape和交换通道的手段也要不一样,你可以自由定义你存储的数据的格式。但在课程中,我们用的数据是别人定义的,所以我们要遵循他们的规则。但要注意,这不是固定的。

1
1
慕标0697996
非常感谢!
2019-05-28
共1条回复

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学,深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程