tensorflow axis

# @Time : 2019/9/2 15:52 # @Author : henu_wqt # @Email : [email protected] # @File : test.py # @Software: PyCharm # !/usr/bin/env python3 # -*- coding: utf-8 -*- from keras import backend as K import numpy as np import tensorflow as tf with tf.Session() as sess: val = [[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]] kvar = K.variable(value=val,dtype='float64', name='example_var') sess.run(tf.global_variables_initializer()) # print(K.dtype(kvar)) print(kvar) # print(K.int_shape(kvar)) print(kvar.eval()) res0 = K.sum(kvar, axis=[0]) print("axis=[0]：\n" ,res0.eval()) res1 = K.sum(kvar, axis=[1]) print("axis=[1]:\n",res1.eval()) res2 = K.sum(kvar, axis=[2]) print("axis=[2]:\n",res2.eval()) res3 = K.sum(kvar, axis=[-1]) print("axis=[-1]:\n",res3.eval()) res4 = K.sum(kvar, axis=[-2]) print("axis=[-2]:\n", res4.eval()) res5 = K.sum(kvar, axis=[-3]) print("axis=[-3]:\n", res5.eval()) res12 = K.sum(kvar, axis=[1,2]) print("axis=[1,2]:\n",res12.eval())

... Using TensorFlow backend ... [[[ 1.  2.  3.]   [ 4.  5.  6.]]  [[ 7.  8.  9.]   [10. 11. 12.]]] axis=[0]：  [[ 8. 10. 12.]  [14. 16. 18.]] axis=[1]:  [[ 5.  7.  9.]  [17. 19. 21.]] axis=[2]:  [[ 6. 15.]  [24. 33.]] axis=[-1]:  [[ 6. 15.]  [24. 33.]] axis=[-2]:  [[ 5.  7.  9.]  [17. 19. 21.]] axis=[-3]:  [[ 8. 10. 12.]  [14. 16. 18.]] axis=[1,2]:  [21. 57.] Process finished with exit code 0

axis：0--- -3     1----  -2      2------ -1