如何批量种植花is？

Question

我有一个张量表示一个图像和几个rois，现在我想将它们裁剪掉。我可以使用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])