torch.fx.experimental.graph_gradual_typechecker.adaptiveavgpool2d_inference_rule#

torch.fx.experimental.graph_gradual_typechecker.adaptiveavgpool2d_inference_rule(n, module_instance)[source]#

The input and output sizes should be the same except for the last two dimensions taken from the input, which represent width and height