Solved by Nikronic in post #2 
              
              
                Hi, 
Have you tried Kornia which supports many different augmentations on both images and tensors. 
For instance, your case be achieved this way: 
import kornia as K

x = torch.randint(0, 1, (60, 40, 170, 128)).float()
K.center_crop(x, (90, 90)).shape

Bests