Pobiera z bariery zadaną liczbę ukończonych elementów.
Ta operacja łączy tensory składowych ukończonych elementów wzdłuż wymiaru zerowego, tworząc tensor pojedynczej składowej.
Elementy wychodzą z bariery, gdy są kompletne i w kolejności, w jakiej zostały umieszczone w barierze. Dane wyjściowe indeksów dostarczają informacji o partii, w której każdy element został pierwotnie wstawiony do bariery.
Klasy zagnieżdżone
klasa | BarieraTakeMany.Options | Opcjonalne atrybuty dla BarrierTakeMany |
Stałe
Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
statyczna barieraTakeMany.Options | zezwolenie na małą partię (wartość logiczna zezwolenia na małą partię) |
statyczna barieraTakeMany | |
Dane wyjściowe <TInt64> | indeksy () Jednowymiarowy tensor indeksów o długości num_elems. |
Dane wyjściowe <TString> | Klucze () Jednowymiarowy tensor kluczy o długości num_elements. |
statyczna barieraTakeMany.Options | timeoutMs (Długi limit czasuMs) |
Lista< Wyjście <?>> | wartości () Jeden wielowymiarowy tensor na komponent w elemencie barierowym. |
statyczna barieraTakeMany.Options | waitForIncomplete (Boolean waitForIncomplete) |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
public static BarrierTakeMany.Options zezwolenieSmallBatch (Boolean zezwolenieSmallBatch)
Parametry
zezwól na małą partię | Zezwalaj na zwrócenie mniej niż num_elements elementów, jeśli bariera jest już zamknięta. |
---|
public static BarrierTakeMany create ( Zakres zakresu, Operand < TString > uchwyt, Operand < TInt32 > numElements, List<Class<? Extends TType >> ComponentTypes, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację BarrierTakeMany.
Parametry
zakres | aktualny zakres |
---|---|
uchwyt | Uchwyt do bariery. |
liczba elementów | Tensor jednoelementowy zawierający liczbę elementów do pobrania. |
Typy komponentów | Typ każdego składnika w wartości. |
opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja BarrierTakeMany
publiczne wyjście < TInt64 > indeksy ()
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 (zaczynając od MIN_LONG i zwiększając się z każdym BarrierInsertMany).
public static BarrierTakeMany.Options timeoutMs (Długi limit czasuMs)
Parametry
przekroczenie limitu czasuPani | Jeśli kolejka jest pusta, ta operacja zostanie zablokowana na maksymalnie timeout_ms milisekund. Uwaga: ta opcja nie jest jeszcze obsługiwana. |
---|
publiczna Lista< Wyjście <?>> wartości ()
Jeden wielowymiarowy tensor na komponent w elemencie barierowym. Wszystkie wartości mają długość num_elements w zerowym wymiarze.