tensoreflusso:: ops:: BarrieraPrendiMolti

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.

Resi:

• 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.

Attributi pubblici

::tensorflow::Output indices

::tensorflow::Output keys

Operation operation

::tensorflow::OutputList values

 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
)

Attrs AllowSmallBatch(
  bool x
)

Attrs TimeoutMs(
  int64 x
)

Attrs WaitForIncomplete(
  bool x
)