jax.lax.reduce_window#

jax.lax.reduce_window(operand, init_value, computation, window_dimensions, window_strides, padding, base_dilation=None, window_dilation=None)[源代码]#

包装 XLA 的 ReduceWindowWithGeneralPadding 运算符。

参数:

  • computation (Callable)

  • window_dimensions (core.Shape)

  • window_strides (Sequence[int])

  • padding（填充） (str | Sequence[tuple[int, int]])

  • base_dilation（基本扩张/膨胀） (Sequence[int] | None | None)

  • window_dilation（窗口扩张/膨胀） (Sequence[int] | None | None)

返回类型:

数组