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