kd.typing.XArray

kd.typing.XArray#

class kauldron.typing.XArray[source]

Bases: object

array_types: tuple[Any, ...] = (<class 'tensorflow.python.framework.tensor.Tensor'>, <class 'numpy.ndarray'>, <class 'jax.Array'>)
dtype

alias of jaxtyping.Shaped