TPUEmbeddingActivations

publiczna klasa końcowa TPUEmbeddingActivations

Opcja umożliwiająca różnicowanie osadzania TPU.

Ta operacja po prostu zwraca pierwsze dane wejściowe, które, jak się zakłada, zostały wycięte z tensorów zwróconych przez TPUEmbeddingDequeueActivations. Obecność tej opcji i jej pierwszego argumentu będącego zmienną, którą można wytrenować, umożliwia automatyczne różnicowanie wykresów zawierających osady za pośrednictwem bibliotek języka Python Embedding TPU.

Metody publiczne

Wyjście <Pływak>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczne aktywacje osadzania TPU
utwórz ( Zakres zakresu, Operand <Float> embeddingVariable, Operand <Float> plasterkowaneActivations, Długi identyfikator tabeli, Długi identyfikator wyszukiwania)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUEmbeddingActivations.
Wyjście <Pływak>

Metody dziedziczone

Metody publiczne

publiczne wyjście <Float> asOutput ()

Zwraca symboliczny uchwyt tensora.

Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.

public static TPUEmbeddingActivations create ( Zakres zakresu, Operand <Float> embeddingVariable, Operand <Float> plasterkowaneAktywacje, Długi identyfikator tabeli, Długi identyfikator wyszukiwania)

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

Parametry
zakres aktualny zakres
osadzaniezmiennej Zmienna możliwa do wyszkolenia, umożliwiająca optymalizatorom znalezienie tej operacji.
pokrojone aktywacje Aktywacje osadzania Tensor powrócą.
identyfikator tabeli Identyfikator tabeli w konfiguracji warstwy osadzania, z której obliczono te aktywacje.
identyfikator wyszukiwania Identyfikator zestawu wskaźników osadzania, który wytworzył te aktywacje.
Zwroty
  • nowa instancja TPUEmbeddingActivations

publiczne wyjście <Float> wyjście ()