array_typing.jax

Functions:

• is_array_like –
• is_jax –

Attributes:

• ArrayLike –
• B –
• B33 –
• B33Like –
• B34 –
• B34Like –
• B43 –
• B43Like –
• B44 –
• B44Like –
• BLike –
• BMN –
• BMNLike –
• BN –
• BN3 –
• BN3Like –
• BN4 –
• BN4Like –
• BNLike –
• BNN –
• BNNLike –
• F –
• F33 –
• F33Like –
• F34 –
• F34Like –
• F43 –
• F43Like –
• F44 –
• F44Like –
• FLike –
• FMN –
• FMNLike –
• FN –
• FN3 –
• FN3Like –
• FN4 –
• FN4Like –
• FNLike –
• FNN –
• FNNLike –
• I –
• I33 –
• I33Like –
• I34 –
• I34Like –
• I43 –
• I43Like –
• I44 –
• I44Like –
• ILike –
• IMN –
• IMNLike –
• IN –
• IN3 –
• IN3Like –
• IN4 –
• IN4Like –
• INLike –
• INN –
• INNLike –

ArrayLike module-attribute

ArrayLike = ArrayLike

B module-attribute

B = Bool[Array, '']

B33 module-attribute

B33 = Bool[Array, '3 3']

B33Like module-attribute

B33Like = Bool[ArrayLike, '3 3']

B34 module-attribute

B34 = Bool[Array, '3 4']

B34Like module-attribute

B34Like = Bool[ArrayLike, '3 4']

B43 module-attribute

B43 = Bool[Array, '4 3']

B43Like module-attribute

B43Like = Bool[ArrayLike, '4 3']

B44 module-attribute

B44 = Bool[Array, '4 4']

B44Like module-attribute

B44Like = Bool[ArrayLike, '4 4']

BLike module-attribute

BLike = Bool[ArrayLike, '']

BMN module-attribute

BMN = Bool[Array, 'M N']

BMNLike module-attribute

BMNLike = Bool[ArrayLike, 'M N']

BN module-attribute

BN = Bool[Array, 'N']

BN3 module-attribute

BN3 = Bool[Array, 'N 3']

BN3Like module-attribute

BN3Like = Bool[ArrayLike, 'N 3']

BN4 module-attribute

BN4 = Bool[Array, 'N 4']

BN4Like module-attribute

BN4Like = Bool[ArrayLike, 'N 4']

BNLike module-attribute

BNLike = Bool[ArrayLike, 'N']

BNN module-attribute

BNN = Bool[Array, 'N N']

BNNLike module-attribute

BNNLike = Bool[ArrayLike, 'N N']

F module-attribute

F = Float[Array, '']

F33 module-attribute

F33 = Float[Array, '3 3']

F33Like module-attribute

F33Like = Float[ArrayLike, '3 3']

F34 module-attribute

F34 = Float[Array, '3 4']

F34Like module-attribute

F34Like = Float[ArrayLike, '3 4']

F43 module-attribute

F43 = Float[Array, '4 3']

F43Like module-attribute

F43Like = Float[ArrayLike, '4 3']

F44 module-attribute

F44 = Float[Array, '4 4']

F44Like module-attribute

F44Like = Float[ArrayLike, '4 4']

FLike module-attribute

FLike = Float[ArrayLike, '']

FMN module-attribute

FMN = Float[Array, 'M N']

FMNLike module-attribute

FMNLike = Float[ArrayLike, 'M N']

FN module-attribute

FN = Float[Array, 'N']

FN3 module-attribute

FN3 = Float[Array, 'N 3']

FN3Like module-attribute

FN3Like = Float[ArrayLike, 'N 3']

FN4 module-attribute

FN4 = Float[Array, 'N 4']

FN4Like module-attribute

FN4Like = Float[ArrayLike, 'N 4']

FNLike module-attribute

FNLike = Float[ArrayLike, 'N']

FNN module-attribute

FNN = Float[Array, 'N N']

FNNLike module-attribute

FNNLike = Float[ArrayLike, 'N N']

I module-attribute

I = Integer[Array, '']

I33 module-attribute

I33 = Integer[Array, '3 3']

I33Like module-attribute

I33Like = Integer[ArrayLike, '3 3']

I34 module-attribute

I34 = Integer[Array, '3 4']

I34Like module-attribute

I34Like = Integer[ArrayLike, '3 4']

I43 module-attribute

I43 = Integer[Array, '4 3']

I43Like module-attribute

I43Like = Integer[ArrayLike, '4 3']

I44 module-attribute

I44 = Integer[Array, '4 4']

I44Like module-attribute

I44Like = Integer[ArrayLike, '4 4']

ILike module-attribute

ILike = Integer[ArrayLike, '']

IMN module-attribute

IMN = Integer[Array, 'M N']

IMNLike module-attribute

IMNLike = Integer[ArrayLike, 'M N']

IN module-attribute

IN = Integer[Array, 'N']

IN3 module-attribute

IN3 = Integer[Array, 'N 3']

IN3Like module-attribute

IN3Like = Integer[ArrayLike, 'N 3']

IN4 module-attribute

IN4 = Integer[Array, 'N 4']

IN4Like module-attribute

IN4Like = Integer[ArrayLike, 'N 4']

INLike module-attribute

INLike = Integer[ArrayLike, 'N']

INN module-attribute

INN = Integer[Array, 'N N']

INNLike module-attribute

INNLike = Integer[ArrayLike, 'N N']

is_array_like

is_array_like(obj: Any) -> bool

Source code in src/array_typing/array_like/_utils.py

def is_array_like(obj: Any) -> bool:
    return hasattr(obj, "__len__") and not isinstance(obj, str | bytes)

is_jax

is_jax(obj: Any) -> TypeGuard[Array]

Source code in src/array_typing/jax/_utils.py

def is_jax(obj: Any) -> TypeGuard[jax.Array]:
    return at.is_instance_named_partial(obj, "jax.Array")