Upsample

Upsample the input tensor. Each dimension value of the output tensor is: output_dimension = floor(input_dimension * scale).

PyTorchApple MLX

PyTorch

API: torch.nn.modules.upsampling.Upsample

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.layers.upsample.Upsample

Strategy: Direct Mapping

Official Docs ↗