我想在pytorch中调整3-D RBG张量的大小。我知道如何调整4-D张量的大小,但不幸的是,这种方法不适用于3-D。
输入为:
#input shape: [3, 100, 200] ---> desired output shape: [3, 80, 120]
如果我有一个4D矢量,那就可以了。
#input shape: [2, 3, 100, 200]
out = torch.nn.functional.interpolate(T,size=(100,80), mode='bilinear')
有什么建议吗?预先感谢!