torch.fx.experimental.symbolic_shapes.constrain_unify
- torch.fx.experimental.symbolic_shapes.constrain_unify(a, b)[source]
Given two SymInts, constrain them so that they must be equal. NB: this will not work with SymInts that represent nontrivial expressions (yet!)