jax.numpy.pad

内容

jax.numpy.pad#

jax.numpy.pad(array, pad_width, mode='constant', **kwargs)[source]#

填充数组。

numpy.pad() 的 LAX 后端实现。

与 NumPy 不同，JAX 的“函数”模式的参数（另一个函数）应该返回修改后的数组。这是因为 Jax 数组是不可变的。（在 NumPy 中，“函数”模式的参数应该就地修改秩 1 数组。）

原始文档字符串如下。

参数：

  • array (array_like of rank N) – 要填充的数组。

  • pad_width ({sequence, array_like, int}) – 填充到每个轴边缘的值的数量。 ((before_1, after_1), ... (before_N, after_N)) 每个轴的唯一填充宽度。 (before, after) 或 ((before, after),) 为每个轴产生相同的填充前和填充后。 (pad,) 或 int 是所有轴的填充前 = 填充后 = 填充宽度的快捷方式。

  • mode (str or function, optional) –

    以下字符串值之一或用户提供的函数。

    ’constant’ (默认)

    用常数值填充。

    ’edge’

    用数组的边缘值填充。

    ’linear_ramp’

    用 end_value 和数组边缘值之间的线性斜坡填充。

    ’maximum’

    用每个轴上所有或部分向量的最大值填充。

    ’mean’

    用每个轴上所有或部分向量的平均值填充。

    ’median’

    沿每个轴使用所有或部分向量的中值进行填充。

    ’minimum’

    沿每个轴使用所有或部分向量的最小值进行填充。

    ’reflect’

    沿每个轴使用向量在向量首尾值上镜像反射的结果进行填充。

    ’symmetric’

    沿数组边缘使用向量镜像反射的结果进行填充。

    ’wrap’

    沿轴使用向量的循环填充。使用第一个值填充末尾，使用末尾值填充开头。

    ’empty’

    使用未定义的值进行填充。

  • stat_length (序列 或 整数, 可选) –

    用于 ‘maximum’、’mean’、’median’ 和 ‘minimum’。每个轴边缘用于计算统计值的值的数量。

    ((before_1, after_1), ... (before_N, after_N)) 每个轴的唯一统计长度。

    (before, after) 或 ((before, after),) 为每个轴产生相同的 before 和 after 统计长度。

    (stat_length,) 或 整数 是 before = after = 统计 长度在所有轴上都相同的一种简写形式。

    默认为 None，使用整个轴。

  • constant_values (序列 或 标量, 可选) –

    用于 ‘constant’。为每个轴设置填充值的值。

    ((before_1, after_1), ... (before_N, after_N)) 每个轴的唯一填充常量。

    (before, after) 或 ((before, after),) 为每个轴产生相同的 before 和 after 常量。

    (constant,) 或 constant 是 before = after = constant 在所有轴上都相同的一种简写形式。

    默认为 0。

  • end_values (序列 或 标量, 可选) –

    用于 ‘linear_ramp’。用于线性斜坡的结束值，并且将形成填充数组的边缘。

    ((before_1, after_1), ... (before_N, after_N)) 每个轴的唯一结束值。

    (before, after) 或 ((before, after),) 为每个轴产生相同的 before 和 after 结束值。

    (constant,) 或 constant 是 before = after = constant 在所有轴上都相同的一种简写形式。

    默认为 0。

  • reflect_type ({'even', 'odd'}, 可选) – 用于 ‘reflect’ 和 ‘symmetric’。’even’ 样式是默认样式，在边缘值周围进行未修改的反射。对于 ‘odd’ 样式，数组的扩展部分是通过从边缘值的二倍值中减去反射值创建的。

返回值:

pad – 等于 array 的秩的填充数组，其形状根据 pad_width 增加。

返回类型:

ndarray

内容