modified augmentation
more changes in augmentation
new aug layer can output its parameters
yet more changes to augmentation: now should be very similar to what was in MATLAB