jax.experimental.pallas.store

内容

jax.experimental.pallas.store(x_ref_or_view, idx, val, *, mask=None, eviction_policy=None)[source]#

在给定索引处存储值。

有关参数含义，请参见 load()。