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() -> 空的 frozenset 对象 frozenset(iterable) -> frozenset 对象

nan_checks

frozenset() -> 空的 frozenset 对象 frozenset(iterable) -> frozenset 对象

index_checks

frozenset() -> 空的 frozenset 对象 frozenset(iterable) -> frozenset 对象

div_checks

frozenset() -> 空的 frozenset 对象 frozenset(iterable) -> frozenset 对象

float_checks

frozenset() -> 空的 frozenset 对象 frozenset(iterable) -> frozenset 对象

automatic_checks

frozenset() -> 空的 frozenset 对象 frozenset(iterable) -> frozenset 对象

all_checks

frozenset() -> 空的 frozenset 对象 frozenset(iterable) -> frozenset 对象