torchvision.transforms flatten

39

import torchvision.transforms as T
transform=T.Compose([T.ToTensor(), T.Lambda(lambda x: torch.flatten(x))])

Comments

Submit
0 Comments