jax.lax.pad

jax.lax.pad(operand, padding_value, padding_config)

对数组应用低、高和/或内部填充。

封装了 XLA 的 Pad 操作符。

参数:

• operand (ArrayLike) – 一个需要填充的数组。

• padding_value (ArrayLike) – 要插入的填充值。必须与 operand 具有相同的 dtype。

• padding_config (Sequence[tuple[int, int, int]]) – 一系列 (低, 高, 内部) 整数元组，给出每个维度中要插入的低、高和内部（膨胀）填充的数量。

返回:

根据 padding_config 在每个维度插入填充值 padding_value 的 operand 数组。

返回类型:

Array