RetrieveAllTPUEmbeddingParameters

publiczna klasa końcowa RetrieveAllTPUEmbeddingParameters

Operacja pobierająca parametry optymalizacyjne z osadzania w pamięci hosta.

Operacja pobierająca parametry optymalizacyjne z osadzania w pamięci hosta. Musi być poprzedzony operacją ConfigureTPUEmbeddingHost, która konfiguruje poprawną konfigurację tabeli osadzania. Na przykład ta operacja służy do pobierania zaktualizowanych parametrów przed zapisaniem punktu kontrolnego. Dla Adagradu, pomocniczy1 będzie zawierał akumulatory po uruchomieniu tej operacji. W przypadku SGD wszystkie wartości pomocnicze* będą puste (tensory 0x0 dla tej tabeli). W przypadku FTRL, pomocniczy1 będzie zawierał akumulatory, a pomocniczy2 będzie zawierał składniki liniowe. W przypadku programu ADAM, pomocniczy1 będzie zawierał pędy, a pomocniczy2 będzie zawierał prędkości.

Metody publiczne

Lista< Wyjście <Float>>
pomocniczy1 ()
Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca pierwszy zapisany pomocniczy parametr optymalizacyjny.
Lista< Wyjście <Float>>
pomocniczy2 ()
Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca drugi przechowywany pomocniczy parametr optymalizacyjny.
Lista< Wyjście <Float>>
pomocniczy3 ()
Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca trzeci przechowywany pomocniczy parametr optymalizacyjny.
Lista< Wyjście <Float>>
pomocniczy4 ()
Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca przechowywany czwarty pomocniczy parametr optymalizacyjny.
Lista< Wyjście <Float>>
pomocniczy5 ()
Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca piąty zapisany pomocniczy parametr optymalizacyjny.
Lista< Wyjście <Float>>
pomocniczy6 ()
Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca sześć przechowywanych pomocniczych parametrów optymalizacyjnych.
Lista< Wyjście <Float>>
pomocniczy7 ()
Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca przechowywany siódmy pomocniczy parametr optymalizacyjny.
statyczne parametry RetrieveAllTPUEmbedding
utwórz (zakres zakresu , długie NumTables, konfiguracja ciągu, długie numShards, długie shardId)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację RetrieveAllTPUEmbeddingParameters.
Lista< Wyjście <Float>>
parametry ()
Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca zapisane parametry tabeli osadzania.

Metody dziedziczone

Metody publiczne

publiczna Lista< Wyjście <Float>> pomocnicza1 ()

Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca pierwszy zapisany pomocniczy parametr optymalizacyjny. Elementy są obecne na liście, ale mają zerowy rozmiar dla niewykorzystanych parametrów optymalizacyjnych (w oparciu o algorytm używany dla każdej tabeli).

public List< Output <Float>> aux2 ()

Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca drugi przechowywany pomocniczy parametr optymalizacyjny. Elementy są obecne na liście, ale mają zerowy rozmiar dla niewykorzystanych parametrów optymalizacyjnych (w oparciu o algorytm używany dla każdej tabeli).

publiczna Lista< Wyjście <Float>> pomocnicza3 ()

Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca trzeci przechowywany pomocniczy parametr optymalizacyjny. Elementy są obecne na liście, ale mają zerowy rozmiar dla niewykorzystanych parametrów optymalizacyjnych (w oparciu o algorytm używany dla każdej tabeli).

publiczna Lista< Wyjście <Float>> pomocnicza4 ()

Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca przechowywany czwarty pomocniczy parametr optymalizacyjny. Elementy są obecne na liście, ale mają zerowy rozmiar dla niewykorzystanych parametrów optymalizacyjnych (w oparciu o algorytm używany dla każdej tabeli).

publiczna Lista< Wyjście <Float>> pomocnicza5 ()

Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca piąty zapisany pomocniczy parametr optymalizacyjny. Elementy są obecne na liście, ale mają zerowy rozmiar dla niewykorzystanych parametrów optymalizacyjnych (w oparciu o algorytm używany dla każdej tabeli).

publiczna Lista< Wyjście <Float>> pomocnicza6 ()

Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca sześć przechowywanych pomocniczych parametrów optymalizacyjnych. Elementy są obecne na liście, ale mają zerowy rozmiar dla niewykorzystanych parametrów optymalizacyjnych (w oparciu o algorytm używany dla każdej tabeli).

publiczna Lista< Wyjście <Float>> pomocnicza7 ()

Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca przechowywany siódmy pomocniczy parametr optymalizacyjny. Elementy są obecne na liście, ale mają zerowy rozmiar dla niewykorzystanych parametrów optymalizacyjnych (w oparciu o algorytm używany dla każdej tabeli).

public static RetrieveAllTPUEmbeddingParameters create (zakres zakresu , długie tabele NumTables, konfiguracja ciągu, długie numShards, długi identyfikator fragmentu)

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

Parametry
zakres aktualny zakres
Liczba tabel Liczba tabel osadzania.
konfiguracja Proto TPUEmbeddingConfiguration opisujące ładowane parametry tabeli, serializowane do ciągu.
liczbaodłamków Liczba fragmentów, na które podzielone są tabele osadzania.
identyfikator fragmentu Identyfikator fragmentu dla tej operacji.
Zwroty
  • nowa instancja RetrieveAllTPUEmbeddingParameters

parametry publiczne Lista< Wyjście <Float>> ()

Lista tensorów, po jednym dla każdej tabeli osadzania, zawierająca zapisane parametry tabeli osadzania.