RandomSelect.py 文件源码

def randomSelectIndex(fromCount, n):
    with tf.name_scope("randomSelectIndex"):
        n = tf.minimum(fromCount, n)
        i = tf.random_shuffle(tf.range(fromCount, dtype=tf.int32))[0:n]
        return tf.expand_dims(i,-1)