View Source Nx.Stream protocol (Nx v0.4.1)

The protocol for streaming data in and out of backends.

Link to this section Summary

Functions

Returns the output of the stream.

Receives data from the stream.

Sends a tensor.

Link to this section Types

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.