Изменение размера RGB Tensor pytorch

Question

Я хочу изменить размер трехмерного тензора RBG в pytorch. Я знаю, как изменить размер 4-D тензора, но, к сожалению, этот метод не работает для 3-D.

Ввод:

#input shape: [3, 100, 200]   ---> desired output shape: [3, 80, 120]

, если у меня есть 4-D вектор все отлично работает.

#input shape: [2, 3, 100, 200]
out = torch.nn.functional.interpolate(T,size=(100,80), mode='bilinear')

Есть предложения? Заранее спасибо!

Answer 1

Благодаря Jodag я нашел ответ:

# input shape [3, 200, 120]
T = T.unsqueeze(0)
T = torch.nn.functional.interpolate(T,size=(100,80), mode='bilinear')
T = T.squeeze(0)
# output shape [3, 100, 80]