EnqueueTPUEmbeddingSparseTensorBatch

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

Stałe

Strunowy OP_NAME Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Metody publiczne

statyczny EnqueueTPUEmbeddingSparseTensorBatch.Options
sumatory (List<String> sumatory)
static <T rozszerza TNumber , U rozszerza TNumber , V rozszerza TNumber > EnqueueTPUEmbeddingSparseTensorBatch
utwórz ( Zakres zakresu, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > 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)

Metody dziedziczone

Stałe

publiczny statyczny końcowy ciąg znaków OP_NAME

Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Wartość stała: „EnqueueTPUEmbeddingSparseTensorBatch”

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

public static EnqueueTPUEmbeddingSparseTensorBatch.Options maxSequenceLengths (List<Long> maxSequenceLengths)