jax.Array.to_device# abstract Array.to_device(device, *, stream=None)[源]# 在指定设备上返回数组的副本 参数: device (xc.Device | Sharding) – Device 或 Sharding,创建的数组将被提交到此。 stream (int | Any | None) – 未实现,传递非None值将导致错误。 self (Array) 返回: 置于指定设备上的数组副本。