从tensor,tensorflow中提取列

时间:2017-10-30 19:56:11

标签: python tensorflow slice convolution

你需要从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])

0 个答案:

没有答案