tensorflow::ops::BarrierTakeMany

Preleva il numero specificato di elementi completati da una barriera.

Riepilogo

Questa operazione concatena i tensori dei componenti degli elementi completi lungo la dimensione 0 per creare un singolo tensore di componenti.

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

Argomentazioni:

• ambito: un oggetto Scope
• maniglia: La maniglia di una barriera.
• num_elements: Un tensore a singolo elemento contenente il numero di elementi da prendere.
• component_types: Il tipo di ciascun componente in un valore.

Attributi opzionali (vedere 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, di lunghezza num_elems. Questi indici si riferiscono al batch in cui i valori sono stati inseriti nella barriera (a partire da MIN_LONG e incrementando con ogni BarrierInsertMany ).
• Chiavi Output : un tensore unidimensionale di chiavi, di 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
)