TensorFlow中获取大于零的元素集合

a为tensor

idx = tf.where(a > 0)

output = tf.gather_nd(a, idx)