Skip to content

MaxPool2d

Bases: BaseModule

2-D max pooling layer.

Parameters:

Name Type Description Default
kernel int | tuple[int, int]

Pooling window size.

required
stride int | tuple[int, int]

Pool stride.

required