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

The protocol for streaming data in and out of backends.

Summary

t()

All the types that implement this protocol.

done(stream)

Returns the output of the stream.

recv(stream)

Receives data from the stream.

Sends a tensor.

@type t() :: term()

All the types that implement this protocol.

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.