jax.experimental.checkify 模块

目录

jax.experimental.checkify 模块#

API#

checkify(f[, errors])

fun 中功能化 check 调用,并可选择添加运行时错误检查。

check(pred, msg, *fmt_args, **fmt_kwargs)

检查一个谓词,如果谓词为假,则添加一个带有消息的错误。

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 对象