torch.fx.experimental.symbolic_shapes.constrain_unify

torch.fx.experimental.symbolic_shapes.constrain_unify(a, b) [source]

Принимает два SymInt и накладывает на них ограничение равенства. Примечание: пока это не работает с SymInt, представляющими нетривиальные выражения.