tensoreflusso:: ops:: BarrieraPrendiMolti

#include <data_flow_ops.h>

Prende il numero indicato di elementi completati da una barriera.

Riepilogo

Questa operazione concatena i tensori dei componenti degli elementi completati lungo la dimensione 0 per creare un tensore a componente singolo.

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

Argomenti:

  • scope: un oggetto Scope
  • 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 ciascun componente in un valore.

Attributi facoltativi (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 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 (a partire da MIN_LONG e aumentando con ogni BarrierInsertMany ).
  • Chiavi Output : un tensore unidimensionale di chiavi, con lunghezza num_elements.
  • Valori OutputList : un tensore qualsiasi dimensionale 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 pubbliche statiche

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

valori

::tensorflow::OutputList values

Funzioni pubbliche

BarrieraPrendiMolti

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

BarrieraPrendiMolti

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

Funzioni pubbliche statiche

Consenti SmallBatch

Attrs AllowSmallBatch(
  bool x
)

TimeoutMs

Attrs TimeoutMs(
  int64 x
)

WaitForIncomplete

Attrs WaitForIncomplete(
  bool x
)