jax.lax.reduce_window

jax.lax.reduce_window(operand, init_value, computation, window_dimensions, window_strides, padding, base_dilation=None, window_dilation=None)

参数:

• computation (可调用对象)

• window_dimensions (core.Shape)

• window_strides (序列[int])

• padding ( str | 序列[tuple[int, int]])

• base_dilation (序列[int] | None | None)

• window_dilation (序列[int] | None | None)

返回类型:

Array