jax.lax.pad#

jax.lax.pad(operand, padding_value, padding_config)[source]#

Applies low, high, and/or interior padding to an array.

Wraps XLA’s Pad operator.

Parameters
Return type

Array

Returns

The operand array with padding value padding_value inserted in each dimension according to the padding_config.