torch.fx.experimental.migrate_gradual_types.constraint_generator.transpose_inference_rule#

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

Can be considered as a sequence of two index selects, so we generate constraints accordingly