值得一提的是jax.numpy.array() ,我們可以用它來直接將ndarray 轉換為DeviceArray。 ... JAX 提供的折衷方法,是利用x.at[idx].set(y) 這種方式:
確定! 回上一頁