publiczna klasa końcowa EnqueueTPUEmbeddingSparseTensorBatch
Ułatwia przenoszenie kodu korzystającego z funkcji tf.nn.embedding_lookup_sparse().
sample_indices[i], embedding_indices[i] i agregation_weights[i] odpowiadają i-tej funkcji. table_ids[i] wskazuje, którą tabelę osadzania należy sprawdzić w ramach tej funkcji.
Tensory w odpowiednich pozycjach trzech list wejściowych (indices_sample_indices, embedding_indices i agregation_weights) 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ą funkcję.
Klasy zagnieżdżone
| klasa | EnqueueTPUEmbeddingSparseTensorBatch.Options | Opcjonalne atrybuty dla EnqueueTPUEmbeddingSparseTensorBatch | |
Metody publiczne
| statyczny EnqueueTPUEmbeddingSparseTensorBatch.Options | sumatory (List<String> sumatory) |
| static <T rozszerza liczbę, U rozszerza liczbę, V rozszerza liczbę> EnqueueTPUEmbeddingSparseTensorBatch | utwórz ( Zakres zakresu, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tableIds, Opcje... opcje) Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację EnqueueTPUEmbeddingSparseTensorBatch. |
| statyczny EnqueueTPUEmbeddingSparseTensorBatch.Options | urządzenieOrdinal (Długie urządzenieOrdinal) |
| statyczny EnqueueTPUEmbeddingSparseTensorBatch.Options | maxSequenceLengths (Lista<Long> maxSequenceLengths) |
| statyczny EnqueueTPUEmbeddingSparseTensorBatch.Options | numFeatures (List<Long> numFeatures) |
Metody dziedziczone
Metody publiczne
public static EnqueueTPUEmbeddingSparseTensorBatch.Options sumatory (sumatory 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 EnqueueTPUEmbeddingSparseTensorBatch create ( Zakres zakresu, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tableIds, Options.. .opcje )
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację EnqueueTPUEmbeddingSparseTensorBatch.
Parametry
| zakres | aktualny zakres |
|---|---|
| przykładoweindeksy | Lista tensorów rangi 1 określająca przykład szkoleniowy, do którego należą odpowiednie wartości embedding_indices i agregation_weights. Odpowiada sp_ids.indices[:,0] w embedding_lookup_sparse(). |
| osadzanie indeksów | Lista tensorów rangi 1, indeksów do tabel osadzania. Odpowiada sp_ids.values w embedding_lookup_sparse(). |
| agregacjaWagi | Lista tensorów rangi 1 zawierająca przykładowe wagi agregacji dla każdego treningu. Odpowiada sp_weights.values w embedding_lookup_sparse(). |
| 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. |
| identyfikatory tabel | Lista liczb całkowitych określających identyfikator tabeli osadzania (przesunięcie TableDescriptor w TPUEmbeddingConfiguration) w celu wyszukania odpowiednich danych wejściowych. I-te wejście jest sprawdzane za pomocą table_ids[i]. Rozmiar listy table_ids musi być równy rozmiarowi próbek_indices, embedding_indices i agregation_weights. |
| opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja EnqueueTPUEmbeddingSparseTensorBatch
public static EnqueueTPUEmbeddingSparseTensorBatch.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ł. |
|---|