jax.export.SymbolicScope#

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

Identifies a scope for symbolic expressions.

All symbolic expressions that interact (e.g., appear in the argument shapes for one JAX function invocation, or are involved in arithmetic operations) must be from the same scope and must share the same SymbolicScope object.

Holds the constraints on symbolic expressions.

See [the README](https://docs.jax.dev/en/latest/export/shape_poly.html#user-specified-symbolic-constraints) for more details.

Parameters:

constraints_str (Sequence[str]) – A sequence of constraints on symbolic dimension expressions, of the form e1 >= e2 or e1 <= e2 or e1 == e2.

__init__(constraints_str=())[source]#
Parameters:

constraints_str (Sequence[str])

Methods

__init__([constraints_str])