jax.experimental.checkify 模块

API

checkify(f[, errors])	将 fun 中的 check 调用函数化，并可选地添加运行时错误检查。
check(pred, msg, *fmt_args, **fmt_kwargs)	检查谓词，如果谓词为 False，则添加一条包含 msg 的错误。
check_error(error)	如果 error 表示失败，则引发异常。
Error(_pred, _code, _metadata, _payload)
JaxRuntimeError
user_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
nan_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
index_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
div_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
float_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
automatic_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
all_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object