tf.argmax函数

tf.argmax函数

tf.argmax(A,axis=0)

axis=0 求每列对大值的索引 axis=1 求每行最大值索引 似乎与通常的0为行，1为列正好相反！！！

代码示例

import tensorflow as tfimport numpy as npA=np.zeros([3])B=np.zeros([3,1])C=np.array([[1,11,22,3], [22,1,2,1], [11,12,2,121]] )init=tf.global_variables_initializer()with tf.Session() as sess: print(A) print(B) print("A的维度",tf.rank(A).eval()) print("B的维度",tf.rank(B).eval()) print("B的shape",tf.shape(B).eval()) print("C的每列最大值索引",tf.argmax(C,axis=0).eval()) print("C的每行最大值索引",tf.argmax(C,axis=1).eval())

output

[0. 0. 0.][[0.] [0.] [0.]]A的维度 1B的维度 2B的shape [3 1]C的每列最大值索引 [1 2 0 2]C的每行最大值索引 [2 0 3]

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。