take a look at the augmentation api we are developing. It’s all tensor based and tries to match torchvision interface: https://kornia.readthedocs.io/en/latest/augmentation.html 228
More to come: https://github.com/kornia/kornia/issues/312 37