之前曾經提到,DeviceArray 是JAX 自行定義的陣列類別,定義在jax.numpy.DeviceArray,它的角色等同 ... JAX 提供的折衷方法,是利用x.at[idx].set(y) 這種方式:
確定! 回上一頁