如何调整PyTorch张量的大小？

Question

现在我在Pytorch中有一个torch.Tensor大小的(5, 1, 44, 44)。

• 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)

Answer 2

TorchVision transforms.functional.resize() 函数正是您所需要的：

import torchvision.transforms.functional as F

t = torch.randn([5, 1, 44, 44])
t_resized = F.resize(t, 224)

如果您希望使用除双线性以外的其他插值模式，您可以使用 interpolation 参数指定。