View Source EXLA.Backend (EXLA v0.2.2)
A Nx tensor backend for the data kept on the device.
You can directly transfer to this backend by calling
Nx.backend_transfer/2
or Nx.backend_copy/2
. It
allows the following options:
:client
- the client to store the data on. Defaults to the client configured inNx.Defn
, otherwise uses:host
.:device_id
- which device to store it on
To get the data out of the device backend into a regular
tensor, call Nx.backend_transfer/1
(with the device
tensor as the single argument).