如何在tensorflow上的separable_conv2d函数中设置RBG图像的数据集？

Question

tensorflow上的separable_conv2d函数中RBG图像的数据集 train_x 应该如何？

我们是否要按照下图所示组织图像？

例如，如果有10个RGB类型和32 * 32大小的图像，train_x的大小将是（32,32,3 * 10）!!？

Answer 1

separable_conv2d（）有一个data_format参数。如果将其设置为NHWC，则输入应为[10,32,32,3]。如果将其设置为NCHW，则输入应为[10,3,32,32]。