EnqueueTPUEmbeddingRaggedTensorBatch

publiczna klasa końcowa EnqueueTPUEmbeddingRaggedTensorBatch

Ułatwia przenoszenie kodu wykorzystującego funkcję tf.nn.embedding_lookup().

sample_splits[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 dwóch list wejściowych, embedding_indices i aggregation_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 EnqueueTPUEmbeddingRaggedTensorBatch.Options Opcjonalne atrybuty dla EnqueueTPUEmbeddingRaggedTensorBatch

Metody publiczne

statyczny EnqueueTPUEmbeddingRaggedTensorBatch.Options
sumatory (List<String> sumatory)
static <T rozszerza liczbę, U rozszerza liczbę, V rozszerza liczbę> EnqueueTPUEmbeddingRaggedTensorBatch
utwórz ( Zakres zakresu, Iterable< Operand <T>> sampleSplits, 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ę EnqueueTPUEmbeddingRaggedTensorBatch.
statyczny EnqueueTPUEmbeddingRaggedTensorBatch.Options
urządzenieOrdinal (Długie urządzenieOrdinal)
statyczny EnqueueTPUEmbeddingRaggedTensorBatch.Options
maxSequenceLengths (Lista<Long> maxSequenceLengths)
statyczny EnqueueTPUEmbeddingRaggedTensorBatch.Options
numFeatures (List<Long> numFeatures)

Metody dziedziczone

Metody publiczne

public static EnqueueTPUEmbeddingRaggedTensorBatch.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 EnqueueTPUEmbeddingRaggedTensorBatch create ( zakres zakresu , Iterable< Operand <T>> sampleSplits, 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ę EnqueueTPUEmbeddingRaggedTensorBatch.

Parametry
zakres aktualny zakres
próbkiPodziały Lista tensorów rangi 1 określająca punkty przerwania służące do dzielenia indeksów embeddingu i wag_agregacji na wiersze. Odpowiada ids.row_splits w embedding_lookup(), gdy id jest RaggedTensor.
osadzanie indeksów Lista tensorów rangi 1, indeksów do tabel osadzania. Odpowiada ids.values ​​w embedding_lookup(), gdy id jest RaggedTensor.
agregacjaWagi Lista tensorów rangi 1 zawierająca przykładowe wagi agregacji dla każdego treningu. Odpowiada polu wartości RaggedTensor z tymi samymi row_splits co id w embedding_lookup(), gdy id jest RaggedTensor.
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 EnqueueTPUEmbeddingRaggedTensorBatch

public static EnqueueTPUEmbeddingRaggedTensorBatch.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 EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths (List<Long> maxSequenceLengths)

public static EnqueueTPUEmbeddingRaggedTensorBatch.Options numFeatures (List<Long> numFeatures)