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.

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 <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 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 <Float>> iterator ()

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

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