在tensorflow中如何获得最小值索引但零除外

Question

我想获得张量的最小值索引，但该值不是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中实现它，谢谢。

Answer 1

这很可怕，但是有效：

override func viewWillDisappear(_ animated: Bool) {
    super.viewWillDisappear(animated)
    navigationController?.setNavigationBarHidden(true, animated: animated)
}```