为了训练一个基于深度学习的模型，我有一个输入Tensor，大小为[batch_size=32，channels=3，Temporal=16，H=128，w=192]，其中包含一个视频帧。（H=224，w=224）。换句话说，我需要一个大小为[batch_size=32，channels=3，Temporal=16，H=224，w=224]的Tensor。我该怎么做呢？
eimct9ow1#
您可以使用torch.nn.functional.interpolate：
torch.nn.functional.interpolate
import torch.nn.functional as nnf y = nnf.interpolate(x, size=(x.shape[2], 224, 224), mode='trilinear')
1条答案
按热度按时间eimct9ow1#
您可以使用
torch.nn.functional.interpolate：