jax.Array.is_fully_addressable#

抽象 属性 Array.is_fully_addressable: bool[源代码]#

这个数组是否完全可寻址?

如果当前进程可以寻址 Sharding 中命名的所有设备,则 jax.Array 是完全可寻址的。is_fully_addressable 在多进程 JAX 中等同于“is_local”。

请注意,完全复制并不等同于完全可寻址,即,一个完全复制的 jax.Array 可以跨多个主机,但并非完全可寻址。