XlaRecvTPUEmbeddingDeduplicationData

publiczna klasa końcowa XlaRecvTPUEmbeddingDeduplicationData

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
utwórz (zakres zakresu , konfiguracja ciągu)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację XlaRecvTPUEmbeddingDeduplicationData.
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

publiczne wyjście <?> wyjście ()