AE_training.py 文件源码

def std_normalize(batch):
    norm_batch = np.zeros(batch.shape)
    for i in range(len(batch)):
        norm_batch[i] = (batch[i] - np.mean(batch[i])) / np.std(batch[i])
    return norm_batch

# Argument parser. This script expects 2 necessory positional args.