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 | |
Iterator< Operand <Float>> | iterator () |
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
public List< Output <Float>> wyjścia ()
Lista TensorList aktywacji osadzania zawierająca jeden Tensor na tabelę osadzania w modelu.