Hi,
You can use torch.nn.functional.interpolate to the down/up scaling for any tensors:
torch.nn.functional.interpolate
Bests