调整RGB张量pytorch的大小

时间:2020-01-18 17:22:42

标签: python pytorch interpolation tensor

我想在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')

有什么建议吗?预先感谢!

0 个答案:

没有答案