nu = np.array([[[1, 1, 1, 1],
[2, 2, 2, 2],
[3, 3, 3, 3]],
[[1, 1, 1, 1],
[2, 2, 2, 2],
[3, 3, 3, 3]]])
mu = nu.reshape([1,2,3,4])
div = tf.reduce_min(mu, axis=3, keepdims=True)
with tf.Session() as sess:
d = sess.run(div)
运行结果