jax.experimental.checkify.Error

jax.experimental.checkify.Error#

class jax.experimental.checkify.Error(_pred, _code, _metadata, _payload)[source]#
Parameters:
__init__(_pred, _code, _metadata, _payload)#
Parameters:

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)