The function in the corresponding libtorch torchvision.transforms.Compose
transforms are still not completely ported and while some transformations already work on tensors instead of
PIL.Images, I don’t think
Compose was ported yet.
This PR seems related and I’m sure contributions are welcome in case you would be interested to port some functions.
CC @fmassa to correct me in case I’m missing something.