Now I have a torch.Tensor
of size (5, 1, 44, 44)
in Pytorch.
5 = batch size
1 = channel
44= image height
44= image width
and I want to 'resize' it to shape (5, 1, 224, 224)
How can I do that? What functions should I use?
It seems like you are looking for interpolate
(a function in 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)