Operacja, która otrzymuje aktywacje osadzania w TPU.
System TPU wykonuje wyszukiwania i agregacje osadzania określone przez argumenty TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch. Wyniki tych agregacji są widoczne na wykresie Tensorflow jako dane wyjściowe operacji RecvTPUEmbeddingActivations. Ta operacja zwraca listę zawierającą jeden Tensor aktywacji na tabelę określoną w modelu. Na wykresie TPU może znajdować się co najwyżej jedna opcja RecvTPUEmbeddingActivations.
Stałe
Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
statyczne aktywacje RecvTPUEembedding | |
Iterator< Operand < TFloat32 >> | iterator () |
Lista< Wyjście < TFloat32 >> | wyjścia () Lista TensorList aktywacji osadzania zawierająca jeden Tensor na tabelę osadzania w modelu. |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
public static RecvTPUEmbeddingActivations create (zakres zakresu , długa liczba wyjść, konfiguracja ciągu)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację RecvTPUEmbeddingActivations.
Parametry
zakres | aktualny zakres |
---|---|
liczba wyjść | Liczba tensorów aktywacji wyjścia równa liczbie tablic osadzania w modelu. |
konfiguracja | Serializowany proto konfiguracji TPUEmbeddingConfiguration. |
Zwroty
- nowa instancja RecvTPUEmbeddingActivations