View Source Nx.Stream protocol (Nx v0.5.1)
The protocol for streaming data in and out of backends.
Link to this section Summary
Link to this section Types
@type t() :: term()
Link to this section Functions
Returns the output of the stream.
It may be a tensor, a tuple of tensors, or a map of tensors.
Receives data from the stream.
It may be a tensor, a tuple of tensors, or a map of tensors.
Sends a tensor.
Returns the given tensor.