jax.experimental.checkify.索引检查

jax.experimental.checkify.index_checks = frozenset({<class 'jax._src.checkify.OOBError'>})

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

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