tensorflow:: ops:: BarrierTakeMany

Pobiera podaną liczbę ukończonych elementów z bariery.

Streszczenie

Operacja ta łączy tensory składowych elementów ukończonych wzdłuż wymiaru 0, tworząc pojedynczy tensor składowy.

Elementy wychodzą z bariery po ich ukończeniu, w kolejności, w jakiej zostały w niej umieszczone. Dane wyjściowe indeksów zawierają informacje o partii, w której każdy element został pierwotnie włożony do bariery.

Argumenty:

• zakres: obiekt zakresu
• uchwyt: Uchwyt bariery.
• num_elements: Tensor pojedynczego elementu zawierający liczbę elementów do pobrania.
• component_types: Typ każdego komponentu w wartości.

Atrybuty opcjonalne (patrz Attrs ):

• allow_small_batch: Zezwala na zwracanie mniejszej liczby elementów niż num_elements, jeśli bariera jest już zamknięta.
• timeout_ms: Jeśli kolejka jest pusta, ta operacja zostanie zablokowana na czas do timeout_ms milisekund. Uwaga: Ta opcja nie jest jeszcze obsługiwana.

Zwroty:

• Indeksy Output : Jednowymiarowy tensor indeksów o długości num_elems. Indeksy te odnoszą się do partii, w której wartości zostały umieszczone w barierze (rozpoczynając od MIN_LONG i rosnąc z każdym BarrierInsertMany ).
• Klucze Output : Jednowymiarowy tensor kluczy o długości num_elements.
• Wartości OutputList : Jeden tensor o dowolnym wymiarze na każdy składnik w elemencie bariery. Wszystkie wartości mają długość num_elements w wymiarze 0.

Atrybuty publiczne

::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
)