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

The protocol for streaming data in and out of backends.

Summary

Types

t()

All the types that implement this protocol.

Functions

Returns the output of the stream.

Receives data from the stream.

Sends a tensor.

Types

@type t() :: term()

All the types that implement this protocol.

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.