tf.argmin()返回矩阵横列或者纵列的最小值的坐标,取决于第二个参数
参数:
tf.argmin(
input,
axis=None,//0是纵列 1是横列
name=None,
dimension=None,
output_type=tf.int64
)
使用案例:
import tensorflow as tf
x = tf.constant([[0, 2, 1], [2, 1, 3]])
sess = tf.Session()
print(sess.run(tf.argmin(x, 0)))//[0 1 0]
print(sess.run(tf.argmin(x, 1)))//[0 1]