[Possible feature] Allow fusing non-constant scalars through "linear" operations
Linux-cpp-lisp opened this issue · comments
It could be useful to provide some way of marking a Tensor as a scalar and adding it to the propagation logic.
Linux-cpp-lisp / opt_einsum_fx
Einsum optimization using opt_einsum and PyTorch FX graph rewriting