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

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ł.