max_pool2d

Auto-generated from paxml_code_defs

PyTorch

API: torch.nn.functional.max_pool2d
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: optax.tree.max
Strategy: Direct Mapping
Official Docs ↗