我想获得张量的最小值索引，但该值不是0 .

a = np.array([[0, 3, 9, 0],
            [0, 0, 5, 7]])
tensor_a = tf.constant(a, dtype=tf.int32)
max_index = tf.argmax(tensor_a, axis=1)

上面的代码定义了一个常量张量，如果我使用tf.argmax，我会得到索引[2,3] . 我如何得到第一行中的3和第二行中的5的索引，最小值但不是零 . 我想得到的真正指数是[1,2] . 如何在tensorflow中实现它，谢谢 .

tensorflow