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 | EnqueueTPUEmbeddingArbitraryTensorBatch.Options | Opcjonalne atrybuty dla EnqueueTPUEmbeddingArbitraryTensorBatch | |
Metody publiczne
| statyczny EnqueueTPUEmbeddingArbitraryTensorBatch.Options | sumatory (List<String> sumatory) |
| static <T rozszerza liczbę, U rozszerza liczbę, V rozszerza liczbę> EnqueueTPUEmbeddingArbitraryTensorBatch | utwórz ( Zakres zasięgu, Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Opcje... opcje) Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację EnqueueTPUEmbeddingArbitraryTensorBatch. |
| statyczny EnqueueTPUEmbeddingArbitraryTensorBatch.Options | urządzenieOrdinal (Długie urządzenieOrdinal) |
Metody dziedziczone
Metody publiczne
public static EnqueueTPUEmbeddingArbitraryTensorBatch.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 EnqueueTPUEmbeddingArbitraryTensorBatch create ( Zakres zakresu, Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację EnqueueTPUEmbeddingArbitraryTensorBatch.
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. |
| opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja EnqueueTPUEmbeddingArbitraryTensorBatch
public static EnqueueTPUEmbeddingArbitraryTensorBatch.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ł. |
|---|