tensorflow之axis

👉🏻阅读模式 | tensorflow | 2018/09/25 | 921次原文发表于：2018-10-16

a = [[[1, 2, 3], [4, 5, 6]],[[1, 2, 3], [4, 5, 6]]]
a[0]

[[1, 2, 3], [4, 5, 6]]

sess = tf.InteractiveSession()
x = tf.constant(a)
x.eval()

array([[[1, 2, 3],
        [4, 5, 6]],

       [[1, 2, 3],
        [4, 5, 6]]], dtype=int32)

tf.reduce_sum(x, axis = 0).eval()

array([[ 2,  4,  6],
       [ 8, 10, 12]], dtype=int32)

tf.reduce_sum(x, axis = 1).eval()

array([[5, 7, 9],
       [5, 7, 9]], dtype=int32)

tf.reduce_sum(x, axis = 2).eval()

array([[ 6, 15],
       [ 6, 15]], dtype=int32)