Sehen Sie sich Keynotes, Produktsitzungen, Workshops und mehr in Google I / O an. Siehe Wiedergabeliste

Tensorflow :: ops :: BarrierTakeMany

#include <data_flow_ops.h>

Nimmt die angegebene Anzahl abgeschlossener Elemente von einer Barriere.

Zusammenfassung

Diese Operation verkettet Komponententensoren mit abgeschlossenen Elementen entlang der 0. Dimension, um einen Einzelkomponententensor zu erstellen.

Elemente kommen aus der Barriere heraus, wenn sie vollständig sind und in der Reihenfolge, in der sie in die Barriere platziert wurden. Die Ausgabe der Indizes enthält Informationen zu der Charge, in die jedes Element ursprünglich in die Barriere eingefügt wurde.

Argumente:

  • scope: Ein Scope- Objekt
  • Griff: Der Griff zu einer Barriere.
  • num_elements: Ein Einzelelement-Tensor, der die Anzahl der zu nehmenden Elemente enthält.
  • component_types: Der Typ jeder Komponente in einem Wert.

Optionale Attribute (siehe Attrs ):

  • allow_small_batch: Erlaubt die Rückgabe von weniger als num_elements Elementen, wenn die Barriere bereits geschlossen ist.
  • timeout_ms: Wenn die Warteschlange leer ist, wird dieser Vorgang für bis zu timeout_ms Millisekunden blockiert. Hinweis: Diese Option wird noch nicht unterstützt.

Kehrt zurück:

  • Output - Indizes: Ein eindimensionaler Tensor - Indizes mit einer Länge num_elems. Diese Indizes beziehen sich auf die Charge, in der die Werte in die Barriere gelegt wurden (beginnend mit MIN_LONG und steigend mit jedem BarrierInsertMany ).
  • Output : Ein eindimensionaler Tensor von Schlüsseln mit der Länge num_elements.
  • OutputList Werte: Ein beliebiger dimensionaler Tensor pro Komponente in einem Barriereelement. Alle Werte haben die Länge num_elements in der 0. Dimension.

Konstruktoren und Destruktoren

BarrierTakeMany (const :: tensorflow::Scope & scope, :: tensorflow::Input handle, :: tensorflow::Input num_elements, const DataTypeSlice & component_types)
BarrierTakeMany (const :: tensorflow::Scope & scope, :: tensorflow::Input handle, :: tensorflow::Input num_elements, const DataTypeSlice & component_types, const BarrierTakeMany::Attrs & attrs)

Öffentliche Attribute

indices
keys
operation
values

Öffentliche statische Funktionen

AllowSmallBatch (bool x)
TimeoutMs (int64 x)
WaitForIncomplete (bool x)

Strukturen

tensorflow :: ops :: BarrierTakeMany :: Attrs

Optionale Attributsetzer für BarrierTakeMany .

Öffentliche Attribute

Indizes

::tensorflow::Output indices

Schlüssel

::tensorflow::Output keys

Operation

Operation operation

Werte

::tensorflow::OutputList values

Öffentliche Funktionen

BarrierTakeMany

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types
)

BarrierTakeMany

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types,
  const BarrierTakeMany::Attrs & attrs
)

Öffentliche statische Funktionen

AllowSmallBatch

Attrs AllowSmallBatch(
  bool x
)

TimeoutMs

Attrs TimeoutMs(
  int64 x
)

WaitForIncomplete

Attrs WaitForIncomplete(
  bool x
)