torch.fx.experimental.migrate_gradual_types.constraint_generator.embedding_inference_rule# torch.fx.experimental.migrate_gradual_types.constraint_generator.embedding_inference_rule(n, module_instance, symbols, constraints, counter)[source]# The output shape differs from the input shape in the last dimension