采用Torchvision的
transforms模块实现: