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