flusso tensoriale:: operazioni:: Barrier Take Many

#include <data_flow_ops.h>

Prende il numero dato di elementi completati da una barriera.

Riepilogo

Questa operazione concatena i tensori dei componenti dell'elemento completato lungo la dimensione 0 per creare un tensore dei componenti singoli.

Gli elementi escono dalla barriera quando sono completi e nell'ordine in cui sono stati inseriti nella barriera. L'output degli indici fornisce informazioni sul batch in cui ciascun elemento è stato originariamente inserito nella barriera.

Argomenti:

  • ambito: un oggetto ambito
  • maniglia: la maniglia di una barriera.
  • num_elements: un tensore a elemento singolo contenente il numero di elementi da prendere.
  • component_types: il tipo di ogni componente in un valore.

Attributi opzionali (vedi Attrs ):

  • allow_small_batch: consente di restituire meno di num_elements elementi se la barriera è già chiusa.
  • timeout_ms: se la coda è vuota, questa operazione si bloccherà per un massimo di timeout_ms millisecondi. Nota: questa opzione non è ancora supportata.

Ritorna:

  • Indici di Output : un tensore unidimensionale di indici, con lunghezza num_elems. Questi indici si riferiscono al batch in cui i valori sono stati inseriti nella barriera (iniziando con MIN_LONG e aumentando con ogni BarrierInsertMany ).
  • Chiavi di Output : un tensore di chiavi unidimensionale, con lunghezza num_elements.
  • Valori OutputList : un tensore qualsiasi dimensione per componente in un elemento barriera. Tutti i valori hanno lunghezza num_elements nella dimensione 0.

Costruttori e Distruttori

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)

Attributi pubblici

indices
keys
operation
values

Funzioni statiche pubbliche

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

Strutture

tensorflow:: ops:: BarrierTakeMany:: Attrs

Setter di attributi facoltativi per BarrierTakeMany .

Attributi pubblici

indici

::tensorflow::Output indices

chiavi

::tensorflow::Output keys

operazione

Operation operation

i valori

::tensorflow::OutputList values

Funzioni pubbliche

Barrier Take Many

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

Barrier Take Many

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

Funzioni statiche pubbliche

Consenti SmallBatch

Attrs AllowSmallBatch(
  bool x
)

TimeoutMs

Attrs TimeoutMs(
  int64 x
)

Attendi per incompleto

Attrs WaitForIncomplete(
  bool x
)