mlx.core.pad#
- mlx.core.pad(a: array, pad_with: Union[int, Tuple[int], Tuple[int, int], List[Tuple[int, int]]], constant_values: Union[scalar, array] = 0, *, stream: Union[None, Stream, Device] = None) array#
Pad an array with a constant value
- Parameters:
a (array) – Input array.
pad_width (int, tuple(int), tuple(int, int) or list(tuple(int, int))) – Number of padded values to add to the edges of each axis:
((before_1, after_1), (before_2, after_2), ..., (before_N, after_N)). If a single pair of integers is passed then(before_i, after_i)are all the same. If a single integer or tuple with a single integer is passed then all axes are extended by the same number on each side.constant_value (array or scalar, optional) – Optional constant value to pad the edges of the array with.
- Returns:
The padded array.
- Return type: