torch.fx.experimental.migrate_gradual_types.constraint_generator.bmm_inference_rule#

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

Constraints that match the input to a size 3 tensor and switch the dimensions according to the rules of batch multiplication