你需要从convo_2
变量中提取列,以便将它们连接成一个张量(每个feature_map中的一列),但我不知道如何做到这一点。我试图使用tf.while_loop()
及其内部tf.slice()
,但我不知道它是否是正确的选择。有人可以提示并向我解释使用tf.slice()
吗?
convo_1 = convolutional_layer(x, shape=[3, 3, 1, 64])
convo_1_pooling = max_pool_2by2(convo_1)
convo_2 = convolutional_layer(convo_1_pooling, shape=[3, 3, 64, 128])