jax.lax.pad

内容

jax.lax.pad#

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

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

包装 XLA 的 Pad 运算符。

参数：:

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

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

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

返回：:

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

返回类型：:

Array

内容