Function infomax_corruption

code/utils.py:237–243

(features, batch)

def infomax_corruption(features, batch):

    if batch is None:
        shuffle_idx = np.random.permutation(features.shape[0])
        shuffle_features = features[shuffle_idx, :]

    return shuffle_features