jax.export.SymbolicScope

class jax.export.SymbolicScope(constraints_str=())

标识符号表达式的范围。

所有交互的符号表达式（例如，出现在一个 JAX 函数调用的参数形状中，或参与算术运算）必须来自同一作用域，并且必须共享同一个 SymbolicScope 对象。

保存符号表达式的约束。

更多详情请参见 [the README](https://jax.readthedocs.io/en/latest/export/shape_poly.html#user-specified-symbolic-constraints)。

参数:

constraints_str (Sequence[str]) – 一系列对符号维度表达式的约束，形式为 e1 >= e2 或 e1 <= e2 或 e1 == e2。

__init__(constraints_str=())

参数:

constraints_str (Sequence[str])

方法

__init__([constraints_str])