Adaptive Instance Normalization Pytorch