Python 如何增加张量的维数？

在下面的代码中，当我打印

conv\u out.get\u 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]

差不多

import numpy as np
np.asarray(conv_out.get_shape().as_list()[1:]).prod()

我应该做这项工作

或者，如果您希望它在tensorflow图的内部，类似于：

tf_shape = tf.shape(conv_out)
tf_shape_prod = tf.reduce_prod(tf_shape[1:])