DynamicEnqueueTPUEmbeddingArbitraryTensorBatch

publiczna klasa końcowa DynamicEnqueueTPUEmbeddingArbitraryTensorBatch

Ułatwia przenoszenie kodu korzystającego z funkcji tf.nn.embedding_lookup_sparse().

embedding_indices[i] i agregation_weights[i] odpowiadają i-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 DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.Options Opcjonalne atrybuty dla DynamicEnqueueTPUEmbeddingArbitraryTensorBatch

Metody publiczne

statyczny DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.Options
sumatory (List<String> sumatory)
static <T rozszerza liczbę, U rozszerza liczbę, V rozszerza liczbę> DynamicEnqueueTPUEmbeddingArbitraryTensorBatch
utwórz ( Zakres zakresu, Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Operand <Integer> urządzenieOrdinal, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.

Metody dziedziczone

Metody publiczne

public static DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.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 DynamicEnqueueTPUEmbeddingArbitraryTensorBatch create ( Zakres zakresu, Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Operand <Integer> urządzenieOrdinal, Opcje.. .opcje )

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.

Parametry
zakres aktualny zakres
sampleIndicesOrRowSplits Lista tensorów rangi 2 określająca przykład szkoleniowy, do którego należą odpowiednie wartości embedding_indices i agregation_weights. Jeśli rozmiar pierwszego wymiaru wynosi 0, zakładamy, że każdy embedding_indices należy do innej próbki. Zarówno int32, jak i int64 są dozwolone i zostaną wewnętrznie przekonwertowane na int32.

Lub lista tensorów rangi 1 określająca podziały wierszy w celu podziału indeksów osadzania i wag_agregacji na wiersze. Odpowiada ids.row_splits w embedding_lookup(), gdy id jest RaggedTensor. Podczas kolejkowania nierównego tensora ND, tylko ostatni wymiar może być nierówny. podział wierszy to gęsty tensor 1-D. Gdy jest pusty, zakładamy, że do op przekazywany jest gęsty tensor. Zarówno int32, jak i int64 są dozwolone i zostaną wewnętrznie przekonwertowane na int32.

osadzanie indeksów Lista tensorów rangi 1, indeksów do tabel osadzania. Zarówno int32, jak i int64 są dozwolone i zostaną wewnętrznie przekonwertowane na int32.
agregacjaWagi Lista tensorów rangi 1 zawierająca przykładowe wagi agregacji dla każdego treningu. Zarówno float32, jak i float64 są dozwolone i zostaną wewnętrznie przekonwertowane na float32.
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.
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ł.
opcje przenosi opcjonalne wartości atrybutów
Zwroty
  • nowa instancja DynamicEnqueueTPUEmbeddingArbitraryTensorBatch