Быстрый способ получить сумму всех значений внутри прямоугольных angular областей интереса карты объектов в pytorch

Question

Поддерживая, что у меня есть тензор N × 4, который является N прямоугольным angular ROI маски (C × H × W), и я хочу получить сумму всех значений каждого канала внутри ROI для получить тензор результата размером N × C, единственный способ, который я могу выяснить, это использовать для l oop. Но этот метод очень медленный, поэтому я хочу спросить, есть ли более быстрый способ сделать это? Метод для l oop выглядит следующим образом:

proposals # N * 4 tensor (ROIs)
mask # C * H * W tensor
total_numbers = []
for i in range(proposals.shape[1]):
    proposal = proposals[i]
    mask_region = mask[:, proposal[0]: proposal[2], proposal[1]: proposal[3]]
    total_number = mask_region.sum(dim=(1, 2))
    total_numbers.append(total_number)