Rate this Page

torch.fx.experimental.migrate_gradual_types.constraint_generator.tensor_inference_rule#

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

If the tensor is a scalar, we will skip it since we do not support scalars yet. We will add support in the future if it’s needed. For our examples so far, scalars are not needed.