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