It looks like you could generate a random tensor, and then add it to the tensor you want to add jitter. If you want more of a random shifting type jitter, I wrote code for that a few days ago here: How to split a (C x H x W) tensor into tiles?
For the rotations, you could probably also use Torch.flip()
, and Torch.diag()
if the rotations are using 90 degree intervals.