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 | |
Metody publiczne
| statyczny EnqueueTPUEmbeddingSparseBatch.Options | sumatory (List<String> sumatory) |
| static <T rozszerza liczbę, U rozszerza liczbę, V rozszerza liczbę> EnqueueTPUEmbeddingSparseBatch | utwórz ( Zakres zakresu, Iterowalny < Operand <T>> sampleIndices, Iterowalny < Operand <U>> embeddingIndices, Iterowalny < Operand <V>> aggregationWeights, Operand <String> 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
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 <String> 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ł. |
|---|