RecvTPUEmbeddingActivations

publiczna klasa końcowa RecvTPUEmbeddingActivations

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

Wartość stała: „Aktywacje RecvTPUEmbedding”

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

publiczny Iterator< Operand < TFloat32 >> iterator ()

public List< Wyjście < TFloat32 >> wyjścia ()

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