jax.lax.conv_with_general_padding

jax.lax.conv_with_general_padding#

jax.lax.conv_with_general_padding(lhs, rhs, window_strides, padding, lhs_dilation, rhs_dilation, precision=None, preferred_element_type=None)[源代码][源代码]#

conv_general_dilated 的便捷包装器。

参数:

  • lhs (Array) – 一个秩为 n+2 维的输入数组。

  • rhs (Array) – 一个秩为 n+2 的核权重数组。

  • window_strides (Sequence[int]) – 一个由 n 个整数组成的序列，表示窗口间的步幅。

  • padding (str | Sequence[tuple[int, int]]) – 字符串 ‘SAME’、字符串 ‘VALID’ 或一个包含 n 个 (low, high) 整数对的序列，这些对分别给出在每个空间维度前后应用的填充。

  • lhs_dilation (Sequence[int] | None) – None，或一个包含 n 个整数的序列，给出在 lhs 的每个空间维度上应用的扩张因子。LHS 扩张也称为转置卷积。

  • rhs_dilation (Sequence[int] | None) – None，或一个包含 n 个整数的序列，给出在 rhs 的每个空间维度上应用的膨胀因子。RHS 膨胀也称为空洞卷积。

  • precision (lax.PrecisionLike) – 可选。可以是 None，这意味着后端的默认精度，一个 Precision 枚举值（Precision.DEFAULT、Precision.HIGH 或 Precision.HIGHEST），或者是一个包含两个 Precision 枚举的元组，指示 lhs` 和 rhs 的精度。

  • preferred_element_type (DTypeLike | None) – 可选。可以是 None，这意味着使用输入类型的默认累积类型，或者是一个数据类型，指示将结果累积到并返回该数据类型的结果。

返回:

包含卷积结果的数组。

返回类型:

Array

目录