jax.export.SymbolicScope

jax.export.SymbolicScope#

class jax.export.SymbolicScope(constraints_str=())[source]#

标识符号表达式的范围。

所有交互的符号表达式(例如,出现在一次 JAX 函数调用中参数形状或参与算术运算)必须来自同一个范围,并必须共享相同的 SymbolicScope 对象。

保存符号表达式上的约束。

有关详细信息,请参阅 [README](https://jax.ac.cn/en/latest/export/shape_poly.html#user-specified-symbolic-constraints)。

参数::

constraints_str (Sequence[str]) – 对符号维度表达式的约束序列,形式为 e1 >= e2e1 <= e2e1 == e2

__init__(constraints_str=())[source]#
参数::

constraints_str (Sequence[str])

方法

__init__([constraints_str])