MaxPooling

diffqc.nn.MaxPooling(x: Array, shape: Tuple[int]) → Array

Max Pooling for 2D

Parameters

• x (jnp.ndarray) – values

• shape (tuple of ints) – shape of taking max. len(shape) == 2.

Returns

maxed values

Return type

jnp.ndarray