tensorflow函数学习(3)

np.prod()函数

用来计算所有元素的乘积，对于有多个维度的数组可以指定轴，如axis=1指定计算每一行的乘积。在神经网络的全连接层利用该函数与resize函数实现多维向量转化为1维向量

In [8]: np.prod([[1,2],[3,4]],0)
Out[8]: array([3, 8])

In [9]: np.prod([[1,2],[3,4]],1)
Out[9]: array([ 2, 12])

In [10]: np.prod([[1,2],[3,4]])
Out[10]: 24

dim = np.prod(prev_layer.get_shape().as_list()[1:])
prev_layer_flat = tf.reshape(prev_layer, [-1, dim])

tf.train.Saver()

sess.run(tf.global_variables_initializer())
saver = tf.train.Saver()
save_path = saver.save(sess, ckpt_path, global_step=1)