Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.graph_gradual_typechecker.flatten_inference_rule#

torch.fx.experimental.graph_gradual_typechecker.flatten_inference_rule(n)[source]#

Applies the flatten shape information to the input then gets the greatest upper bound of the resulting type and the existing type

Return type:

Any

On this page
Show Source
PyTorch Libraries