jax.export.is_symbolic_dim

jax.export.is_symbolic_dim#

jax.export.is_symbolic_dim(p)[source]#

Checks if a dimension is symbolic.

Parameters:

p (DimSize)

Return type:

TypeGuard[_DimExpr]