torch.fx.experimental.migrate_gradual_types.constraint_generator.index_select_inference_rule#

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

We constrain the second argument to a vector or Dyn. The output replaces the input with the shape of the vector at the position given by the index (first argument)