jax.experimental.checkify.浮点检查

jax.experimental.checkify.float_checks = frozenset({<class 'jax._src.checkify.DivisionByZeroError'>, <class 'jax._src.checkify.NaNError'>})

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

构建一个不可变的无序唯一元素集合。