小编R.a*_*ali的帖子

在下面的代码中，当我打印它时，conv_out.get_shape()它会给我输出(1,14,14,1)。我想将第二个第三个和第四个维度相乘(14*14*1)。我怎样才能做到这一点？

input = tf.Variable(tf.random_normal([1,28,28,1]))
filter = tf.Variable(tf.random_normal([5,5,1,1]))

def conv2d(input,filter):
    return tf.nn.conv2d(input,filter,strides=[1,2,2,1],padding='SAME')

conv_out = conv2d(input,filter)
sess = tf.InteractiveSession()
sess.run(tf.initialize_all_variables())

print conv_out.get_shape()
print conv_out.get_shape().as_list()[2]