def flip(inputs, labels, axis): ''' axis : integer. Axis in array, which entries are reversed. ''' return np.flip(inputs, axis), np.flip(labels, axis)