layers.py 文件源码

def K_n_choose_k(n, k, seed=None):
        import tensorflow as tf
        if seed is None:
            seed = np.random.randint(10e6)
        x = tf.range(0, limit=n, dtype='int32')
        x = tf.random_shuffle(x, seed=seed)
        x = x[0:k]
        return x