XlaSparseCoreAdam

publiczna klasa końcowa XlaSparseCoreAdam

Metody publiczne

statyczny XlaSparseCoreAdam
utwórz ( Zakres zakresu, Operand <Float> embeddingTable, Operand <Integer> indeksy, Argument <Float> gradient, Argument <Float> learningRate, Argument <Float> pęd, Argument <Float> prędkość, Argument <Float> beta1, Argument <Float > beta2, Operand <Float> epsilon, długa funkcjaWidth, wartość logiczna useSumInsideSqrt)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację XlaSparseCoreAdam.
Wyjście <Pływak>
Wyjście <Pływak>
Wyjście <Pływak>

Metody dziedziczone

Metody publiczne

public static XlaSparseCoreAdam create ( Zakres zasięgu , Operand <Float> embeddingTable, Operand <Integer> indeksy, Argument <Float> gradient, Argument <Float> learningRate, Operand <Float> pęd, Argument <Float> prędkość, Argument <Float> beta1, Operand <Float> beta2, Operand <Float> epsilon, długa funkcjaWidth, wartość logiczna useSumInsideSqrt)

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

Parametry
zakres aktualny zakres
Zwroty
  • nową instancję XlaSparseCoreAdam

publiczne wyjście <Float> zaktualizowaneEmbeddingTable ()

publiczne wyjście <Float> zaktualizowano Momentum ()

publiczne wyjście <Float> zaktualizowanoPrędkość ()