jax.experimental.checkify.Error

class jax.experimental.checkify.Error(_pred, _code, _metadata, _payload)

Parameters

• _pred (Dict[ErrorEffect, Union[bool, Array]]) –

• _code (Dict[ErrorEffect, Union[int, Array]]) –

• _metadata (Dict[Union[int, Array], PyTreeDef]) –

• _payload (Dict[ErrorEffect, List[Union[ndarray, Array]]]) –

__init__(_pred, _code, _metadata, _payload)

Parameters

• _pred (Dict[ErrorEffect, Union[bool, Array]]) –

• _code (Dict[ErrorEffect, Union[int, Array]]) –

• _metadata (Dict[Union[int, Array], PyTreeDef]) –

• _payload (Dict[ErrorEffect, List[Union[ndarray, Array]]]) –

Methods

__init__(_pred, _code, _metadata, _payload)	param _pred
get()	Returns error message if error happened, None if no error happened.
get_exception()	Returns Python exception if error happened, None if no error happened.
throw()
tree_flatten()
tree_unflatten(metadata, data)