add augment layer
add augment layer test
patch-wise normalization