ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

InfeedDequeueTuple

public final class InfeedDequeueTuple

Fetches multiple values from infeed as an XLA tuple.

Public Methods

static InfeedDequeueTuple
create(Scope scope, List<Class<?>> dtypes, List<Shape> shapes)
Factory method to create a class wrapping a new InfeedDequeueTuple operation.
Iterator<Operand<Object>>
List<Output<?>>
outputs()
A list of tensors that will be provided using the infeed mechanism.

Inherited Methods

Public Methods

public static InfeedDequeueTuple create (Scope scope, List<Class<?>> dtypes, List<Shape> shapes)

Factory method to create a class wrapping a new InfeedDequeueTuple operation.

Parameters
scope current scope
dtypes The element types of each element in `outputs`.
shapes The shapes of each tensor in `outputs`.
Returns
  • a new instance of InfeedDequeueTuple

public Iterator<Operand<Object>> iterator ()

public List<Output<?>> outputs ()

A list of tensors that will be provided using the infeed mechanism.