EnqueueTPUEmbeddingSparseBatch

publiczna klasa końcowa EnqueueTPUEmbeddingSparseBatch

Operacja kolejkująca indeksy wejściowe TPUEmbedding z SparseTensor.

Ta operacja ułatwia przenoszenie kodu korzystającego z funkcji embedding_lookup_sparse(), chociaż w celu wygenerowania argumentów tej operacji wymagane jest wstępne przetwarzanie w Pythonie argumentów SparseTensor do embedding_lookup_sparse(), ponieważ na krok szkolenia dozwolona jest tylko jedna operacja EnqueueTPUEmbeddingSparseBatch.

Tensory w odpowiednich pozycjach na trzech listach wejściowych muszą mieć ten sam kształt, tj. rangę 1 z dim_size() równą całkowitej liczbie odnośników do tabeli opisanej przez odpowiedni table_id.

Klasy zagnieżdżone

klasa

KolejkujTPUEmbeddingSparseBatch.Opcje

Opcjonalne atrybuty dla EnqueueTPUEmbeddingSparseBatch

Stałe

Strunowy

OP_NAME

Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Metody publiczne

statyczny EnqueueTPUEmbeddingSparseBatch.Options	sumatory (List<String> sumatory)
static <T rozszerza TNumber , U rozszerza TNumber , V rozszerza TNumber > EnqueueTPUEmbeddingSparseBatch	utwórz ( Zakres zakresu, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, Opcje... opcje) Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację EnqueueTPUEmbeddingSparseBatch.
statyczny EnqueueTPUEmbeddingSparseBatch.Options	urządzenieOrdinal (Długie urządzenieOrdinal)

Metody dziedziczone

Z klasy org.tensorflow.op.RawOp

końcowa wartość logiczna	równa się (obiekt obiektu)
końcowy wew	hashCode ()
Operacja	op () Zwróć tę jednostkę obliczeniową jako pojedynczą Operation .
ostatni ciąg	doString ()

Z klasy java.lang.Object

wartość logiczna	równa się (Obiekt arg0)
ostatnia klasa<?>	pobierzKlasę ()
wew	hashCode ()
ostateczna pustka	powiadomić ()
ostateczna pustka	powiadom wszystkich ()
Strunowy	doString ()
ostateczna pustka	czekaj (długi arg0, int arg1)
ostateczna pustka	czekaj (długi arg0)
ostateczna pustka	Czekać ()

Z interfejsu org.tensorflow.op.Op

abstrakcyjne środowisko wykonania	środowisko () Zwróć środowisko wykonawcze, w którym utworzono tę operację.
abstrakcyjna operacja	op () Zwróć tę jednostkę obliczeniową jako pojedynczą Operation .

Stałe

publiczny statyczny końcowy ciąg znaków OP_NAME

Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Wartość stała: „EnqueueTPUEmbeddingSparseBatch”

Metody publiczne

publiczne statyczne sumatory EnqueueTPUEmbeddingSparseBatch.Options (łączniki List<String>)

Parametry

kombinatory

Lista skalarów łańcuchowych, po jednym dla każdej tabeli osadzania, która określa sposób normalizacji aktywacji osadzania po sumowaniu ważonym. Obsługiwane sumatory to „średnia”, „suma” lub „sqrtn”. Nieprawidłowe jest, gdy suma wag wynosi 0 dla „średniej” lub suma kwadratów wag wynosi 0 dla „sqrtn”. Jeśli sumatory nie zostaną przekazane, domyślnie dla wszystkich tabel zostanie użyta wartość „suma”.

public static EnqueueTPUEmbeddingSparseBatch create ( Zakres zakresu, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, Opcje... opcje)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację EnqueueTPUEmbeddingSparseBatch.

Parametry

zakres	aktualny zakres
przykładoweindeksy	Lista tensorów rangi 1 określająca przykład szkolenia i funkcję, do której należą odpowiednie wartości embedding_indices i agregation_weights. próbki_indices[i] musi być równe b * nf + f, gdzie nf to liczba cech z odpowiedniej tabeli, f jest w [0, nf), a b jest w [0, wielkość partii).
osadzanie indeksów	Lista tensorów rangi 1, indeksów do tabel osadzania.
agregacjaWagi	Lista tensorów rangi 1 zawierająca na próbkę – tj. na (przykład uczący, cechę) – wagi agregacji.
trybZastąpienie	Ciąg wejściowy, który zastępuje tryb określony w TPUEmbeddingConfiguration. Obsługiwane wartości to {'nieokreślone', 'wnioskowanie', 'trening', 'backward_pass_only'}. W przypadku ustawienia na „unspecified” używany jest tryb ustawiony w TPUEmbeddingConfiguration, w przeciwnym razie używany jest mode_override.
opcje	przenosi opcjonalne wartości atrybutów

Zwroty

• nowa instancja EnqueueTPUEmbeddingSparseBatch

public static EnqueueTPUEmbeddingSparseBatch.Options urządzenieOrdinal (Długie urządzenieOrdinal)

Parametry

urządzeniePorządek	Urządzenie TPU, którego chcesz użyć. Powinna wynosić >= 0 i być mniejsza od liczby rdzeni TPU w zadaniu, na którym umieszczony jest węzeł.