Method _process_sym_expr

source/python.js:20541–20568

(sym, hint)

            }
            _parse_sym_expr(expr_str, hint) {
                const _process_sym_expr = (sym, hint) => {
                    if (sym.is_Integer || sym.is_Float || sym.is_Boolean) {
                        return sym;
                    }
                    expr_str = sym.__str__();
                    for (const arg of sym.args) {
                        this._parse_sym_expr(arg);
                    }
                    if (this.symbol_name_to_symbol.has(expr_str)) {
                        sym = this.symbol_name_to_symbol.get(expr_str);
                    } else {
                        this.symbol_name_to_symbol.set(expr_str, sym);
                        if (builtins.isinstance(sym, sympy.core.symbol.Symbol) && torch.fx.experimental.symbolic_shapes.symbol_is_type(sym, [torch.utils._sympy.symbol.SymT.UNBACKED_INT, torch.utils._sympy.symbol.SymT.UNBACKED_FLOAT])) {
                            this.unbacked_symbols.add(sym);
                        }
                    }
                    if (hint !== null && !this.shape_env.var_to_val.has(sym)) {
                        this.shape_env.add_var_to_val(sym, hint);
                    }
                    const vr = this.symbol_name_to_range.get(expr_str);
                    if (vr) {
                        this.shape_env.constrain_symbol_range(sym, vr.lower, vr.upper);
                    }
                    if (builtins.isinstance(sym, sympy.core.symbol.Symbol)) {
                        this.shape_env.var_to_stack.set(sym, torch.utils._traceback.CapturedTraceback.extract(false, false, 1));
                    }
                    return sym;
                };
                const locals = new Map([...this.sympy_functions, ...this.symbol_name_to_symbol]);
                const expr = sympy.core.sympify.sympify(expr_str, locals);
                return _process_sym_expr(expr, hint);