jax.export.SymbolicScope#
- class jax.export.SymbolicScope(constraints_str=())[源代码]#
标识符号表达式的范围。
所有交互的符号表达式(例如,出现在一个 JAX 函数调用的参数形状中,或者参与算术运算)必须来自相同的范围,并且必须共享相同的 SymbolicScope 对象。
保持符号表达式的约束。
有关更多详细信息,请参阅 [README](https://jax.ac.cn/en/latest/export/shape_poly.html#user-specified-symbolic-constraints)。
- 参数:
constraints_str (Sequence[str]) – 符号维度表达式的约束序列,形式为 e1 >= e2 或 e1 <= e2 或 e1 == e2。
方法
__init__
([constraints_str])