Pytorch normalize example.