XlaRecvTPUEmbeddingActivations

publiczna klasa końcowa XlaRecvTPUEmbeddingActivations

Operacja, która otrzymuje aktywacje osadzania w TPU.

System TPU dokonuje osadzania wyszukiwań i agregacji. Wyniki tych agregacji są widoczne na wykresie Tensorflow jako dane wyjściowe operacji XlaRecvTPUEmbeddingActivations. Ta operacja zwraca listę zawierającą jeden Tensor aktywacji na tabelę określoną w modelu.

Metody publiczne

statyczne aktywacje XlaRecvTPUEembedding
utwórz ( Zakres zakresu, Operand <?> deduplicationData, Long numTables, Konfiguracja ciągu)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację XlaRecvTPUEmbeddingActivations.
Iterator< Operand <Float>>
Lista< Wyjście <Float>>
wyjścia ()
Lista TensorList aktywacji osadzania zawierająca jeden Tensor na tabelę osadzania w modelu.

Metody dziedziczone

Metody publiczne

public static XlaRecvTPUEmbeddingActivations create ( Zakres zakresu, Operand <?> deduplicationData, Long numTables, Konfiguracja ciągu)

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

Parametry
zakres aktualny zakres
deduplikacjaDane Tensor o typie=DT_VARIANT zawierający dane deduplikacyjne. 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.
liczba tabel Liczba tensorów aktywacji wyjścia. Jeśli deskryptor funkcji jest obecny w konfiguracji osadzania tpu, jest on równy liczbie funkcji w przeciwnym razie równej liczbie tabel osadzania w modelu.
konfiguracja Serializowany proto konfiguracji TPUEmbeddingConfiguration.
Zwroty
  • nowa instancja XlaRecvTPUEmbeddingActivations

publiczny Iterator< Operand <Float>> iterator ()

public List< Output <Float>> wyjścia ()

Lista TensorList aktywacji osadzania zawierająca jeden Tensor na tabelę osadzania w modelu.