Как собрать урожай ржи в виде партии?

Question

У меня есть тензор, представляющий изображение, и несколько кадров, теперь я хочу обрезать их. Я могу сделать это с помощью цикла for, но есть ли функция pytorch, которая может сделать это быстрее?

import torch

img = torch.randn(3, 224, 224)
rois = torch.Tensor([[10, 10, 30, 30],
                     [40, 40, 80, 80]])
cropped = []
for roi in rois:
    x1, y1, x2, y2 = roi
    cropped.append(img[:, y1:y2, x1:x2])