Odbiera dane deduplikacyjne (wskaźniki i wagi) z rdzenia osadzającego.
Dane deduplikacyjne to Tensor o typie=DT_VARIANT. Sam tensor jest zagnieżdżoną krotką XLA zawierającą N elementów (gdzie N jest stosunkiem liczby osadzonych do rdzeni tensorowych na chip TPU). Każdy element zagnieżdżonej krotki jest krotką tensorów rangi 1. Każdy tensor zawiera indeksy (DT_UINT32) do wyszukiwania osadzania w TensorCore lub wagi (DT_FLOAT) do zastosowania do danych wyjściowych operacji wyszukiwania osadzania.
Metody publiczne
| Wyjście <Obiekt> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
| statyczne XlaRecvTPUEmbeddingDeduplicationData | |
| Wyjście <?> | wyjście () |
Metody dziedziczone
Metody publiczne
publiczne wyjście <Obiekt> asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
public static XlaRecvTPUEmbeddingDeduplicationData create ( zakres zakresu , konfiguracja ciągu)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację XlaRecvTPUEmbeddingDeduplicationData.
Parametry
| zakres | aktualny zakres |
|---|---|
| konfiguracja | Serializowany proto konfiguracji TPUEmbeddingConfiguration. |
Zwroty
- nowa instancja XlaRecvTPUEmbeddingDeduplicationData