jax.Array.is_fully_addressable

jax.Array.is_fully_addressable#

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

此 Array 是否完全可寻址?

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

请注意,完全复制(fully replicated)并不等于完全可寻址(fully addressable),即一个完全复制的 jax.Array 可能跨越多个主机,因此并非完全可寻址。