Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.migrate_gradual_types.constraint_generator.cumsum_inference_rule#

torch.fx.experimental.migrate_gradual_types.constraint_generator.cumsum_inference_rule(n, symbols, constraints, counter)[source]#

Input and output shapes should be equal We should verify that the index is valid

Return type:

tuple[list[Constraint], int]

On this page
Show Source
PyTorch Libraries