Nx
Nx v0.4.1

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

The protocol for streaming data in and out of backends.

Link to this section Summary

Types

t()

Functions

done(stream)

Returns the output of the stream.

recv(stream)

Receives data from the stream.

send(stream, tensor)

Sends a tensor.

Link to this section Types

Link to this type

t()

View Source 
@type t() :: term()

Link to this section Functions

Link to this function

done(stream)

View Source

Returns the output of the stream.

It may be a tensor, a tuple of tensors, or a map of tensors.

Link to this function

recv(stream)

View Source

Receives data from the stream.

It may be a tensor, a tuple of tensors, or a map of tensors.

Link to this function

send(stream, tensor)

View Source

Sends a tensor.

Returns the given tensor.