See https://github.com/lorenzoh/DataAugmentation.jl/issues/56
See FluxML/DataAugmentation.jl#56