评价此页

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]#

输出形状在最后一个维度上与输入形状不同