EnqueueTPUEmbeddingBatch

publiczna klasa końcowa EnqueueTPUEmbeddingBatch

Operacja kolejkująca listę wejściowych tensorów wsadowych do TPUEmbedding.

Operacja kolejkująca listę wejściowych tensorów wsadowych do TPUEmbedding.

Klasy zagnieżdżone

klasa KolejkujTPUEmbeddingBatch.Opcje Opcjonalne atrybuty dla EnqueueTPUEmbeddingBatch

Metody publiczne

statyczny EnqueueTPUEmbeddingBatch.Options
sumatory (List<String> sumatory)
statyczny EnqueueTPUEmbeddingBatch
utwórz (zakres zakresu , Iterable< Operand <String>> partia, Operand <String> modeOverride, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację EnqueueTPUEmbeddingBatch.
statyczny EnqueueTPUEmbeddingBatch.Options
urządzenieOrdinal (Długie urządzenieOrdinal)

Metody dziedziczone

Metody publiczne

publiczne statyczne sumatory EnqueueTPUEmbeddingBatch.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 EnqueueTPUEmbeddingBatch create ( Zakres zakresu, Iterowalna < Operand <String>> partia, Operand <String> modeOverride, Opcje... opcje)

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

Parametry
zakres aktualny zakres
seria Lista tensorów 1D, po jednym dla każdej tabeli osadzania, zawierająca dane wejściowe wsadowe zakodowane jako protos dist_belief.SparseFeatures. Jeśli pole wagi w proto SparseFeatures nie jest wypełnione dla identyfikatora, przyjmowana jest waga 1,0.
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 EnqueueTPUEmbeddingBatch

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

Parametry
urządzeniePorządek Urządzenie TPU, którego chcesz użyć. Powinno to wynosić -1, gdy Op jest uruchomione na urządzeniu TPU i >= 0, gdy Op działa na urządzeniu z procesorem.