jax.Array.to_device

abstract Array.to_device(device, *, stream=None)

傳回指定裝置上陣列的副本

參數:

• device (xc.Device | Sharding) – 將在其中提交所建立陣列的 Device 或 Sharding。

• stream (int | Any | None) – 尚未實作，傳遞非 None 值將導致錯誤。

• self (Array)

傳回:

放置在指定裝置或裝置上的陣列副本。