TensorSpace.to_device
- TensorSpace.to_device(device: str)[source]
Return a copy of this space with storage on a different computational device. Mathematically this is the same space. It also uses the same backend for array operations.
Parameters
- device :
Where elements of this space store their arrays. The default spaces store on
'cpu'. Which alternatives are possible depends on the backend (impl) and hardware availability.
Returns
- newspace
TensorSpace Version of this space with selected device.