Pad

Given a tensor containing the data to be padded (data), a tensor containing the number of start and end pad values for axis (pads), (optionally) a mode, and (optionally) constant_value, a padded tensor (output) is generated. The three supported modes are (similar to corresponding modes s…

Abstract Signature:

Pad(data: Tensor, pads: int, constant_value: Tensor, axes, mode: str)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNX

PyTorch

API: torch.nn.functional.pad

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.numpy.pad

Strategy: Plugin (padding_converter)

NumPy

API: numpy.pad

Strategy: Plugin (padding_converter)

Keras

API: keras.ops.pad

Strategy: Plugin (padding_converter)

TensorFlow

API: tf.pad

Strategy: Plugin (padding_converter)

Apple MLX

API: mlx.core.pad

Strategy: Plugin (padding_converter)

Flax NNX

API: jax.numpy.pad

Strategy: Plugin (padding_converter)