torch.nn.functional.interpolate用法

torch.nn.functional.interpolate:将图片上/下采样到指定的大小
torch.nn.functional.interpolate(input,size=)
(本次示例输入是4维)
输入的维度:batch_size×channels×height×width
size:output size
代码示例:

import torch
import torch.nn.functional as F
a=torch.randn(2,3,4,5)
b=F.interpolate(a,2)
print('a的shape:',a.shape)
print("b的shape:",b.shape)

输出:

a的shape: torch.Size([2, 3, 4, 5])
b的shape: torch.Size([2, 3, 2, 2])