TensorFlow:获取非零张量中最小元素的索引的有效方法？

import tensorflow as tf

# 创建一个张量
tensor = tf.constant([0, 2, 1, 0, 3, 0, 4])

# 使用tf.where函数找到非零元素的索引
indices = tf.where(tf.not_equal(tensor, 0))

# 使用tf.argmin函数找到最小元素的索引
min_index = tf.argmin(tf.reshape(tensor, [-1]))

# 打印结果
with tf.Session() as sess:
    print("非零元素的索引：", sess.run(indices))
    print("最小元素的索引：", sess.run(min_index))

非零元素的索引： [[1]
 [2]
 [4]
 [6]]
最小元素的索引： 0