jax.experimental.checkify 模块

API

checkify(f[, errors])	功能化 fun 中的 check 调用，并可选择添加运行时错误检查。
check(pred, msg, *fmt_args[, debug])	检查谓词，如果谓词为 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