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