Как изменить размер тензора PyTorch?

Question

Теперь у меня есть torch.Tensor размера (5, 1, 44, 44) в Pytorch.

• 5 = размер пакета
• 1 = канал
• 44 = высота изображения
• 44 = ширина изображения

, и я хочу изменить его размер, чтобы придать ему форму (5, 1, 224, 224)

Как я могу это сделать? Какие функции я должен использовать?

Answer 1

Похоже, вы ищете interpolate (функция в nn.functional):

import torch.nn.functional as nnf

x = torch.rand(5, 1, 44, 44)
out = nnf.interpolate(x, size=(224, 224), mode='bicubic', align_corners=False)