jax.lax.conv#

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

围绕 conv_general_dilated 的便捷包装器。

参数:

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

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

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

  • padding (str) – 字符串 ‘SAME’ 或字符串 ‘VALID’。

  • precision (lax.PrecisionLike | None) – 可选参数。可以是 None，表示使用后端的默认精度；可以是 Precision 枚举值（Precision.DEFAULT、Precision.HIGH 或 Precision.HIGHEST）；或者是一个包含两个 Precision 枚举值的元组，分别表示 lhs 和 rhs 的精度。

  • preferred_element_type (DTypeLike | None | None) – 可选参数。可以是 None，表示使用输入类型的默认累积类型；或者是一个数据类型，表示累积结果并以该数据类型返回结果。

返回:

包含卷积结果的数组。

返回类型:

数组