jax.experimental.checkify.除法检查# jax.experimental.checkify.div_checks = frozenset({<class 'jax._src.checkify.DivisionByZeroError'>})# frozenset() -> 空 frozenset 对象 frozenset(iterable) -> frozenset 对象 构建一个不可变的无序唯一元素集合。