torch.fx.experimental.graph_gradual_typechecker.flatten_refinement_rule

torch.fx.experimental.graph_gradual_typechecker.flatten_refinement_rule(n)

Generates equality constraints between the dimensions of the input and output that will not be involved in the flatten operation

Return type:

list[Any]