import torch import torchvision.transforms as T transforms = T.Compose([ T.ToTensor(),T.Resize((256,256), antialias=False), T.Lambda(lambda x: x.to(torch.float32))])