Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.graph_gradual_typechecker.maxpool2d_check#

torch.fx.experimental.graph_gradual_typechecker.maxpool2d_check(typ, module_instance)[source]#

Applies the maxpool2d shape information to the input this affects the last two dimensions

Return type:

TensorType

On this page
Show Source
PyTorch Libraries