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)

Jax运行时错误

用户检查

frozenset() -> 空的 frozenset 对象 frozenset(可迭代对象) -> frozenset 对象

NaN 检查

frozenset() -> 空的 frozenset 对象 frozenset(可迭代对象) -> frozenset 对象

索引检查

frozenset() -> 空的 frozenset 对象 frozenset(可迭代对象) -> frozenset 对象

除法检查

frozenset() -> 空的 frozenset 对象 frozenset(可迭代对象) -> frozenset 对象

浮点数检查

frozenset() -> 空的 frozenset 对象 frozenset(可迭代对象) -> frozenset 对象

自动检查

frozenset() -> 空的 frozenset 对象 frozenset(可迭代对象) -> frozenset 对象

所有检查

frozenset() -> 空的 frozenset 对象 frozenset(可迭代对象) -> frozenset 对象