InfeedDequeueTuple

Stay organized with collections Save and categorize content based on your preferences.
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.