XlaSparseCoreAdam

публичный финальный класс XlaSparseCoreAdam

Публичные методы

статический XlaSparseCoreAdam
create ( Область видимости , Операнд <Float> embeddingTable, Операнд <Integer> индексы, Операнд <Float> градиент, Операнд <Float> LearningRate, Операнд <Float> импульс, Операнд <Float> скорость, Операнд <Float> beta1, Операнд <Float > beta2, операнд <Float> epsilon, Long FeatureWidth, логическое значение useSumInsideSqrt)
Фабричный метод для создания класса, обертывающего новую операцию XlaSparseCoreAdam.
Выход <С плавающей запятой>
Выход <С плавающей запятой>
Выход <С плавающей запятой>

Унаследованные методы

Публичные методы

public static XlaSparseCoreAdam create ( Область видимости , Операнд <Float> embeddingTable, Операнд <Integer> индексы, Операнд <Float> градиент, Операнд <Float> LearningRate, Операнд <Float> импульс, Операнд <Float> скорость, Операнд <Float> beta1, Операнд <Float> beta2, Операнд <Float> epsilon, Long FeatureWidth, логическое значение useSumInsideSqrt)

Фабричный метод для создания класса, обертывающего новую операцию XlaSparseCoreAdam.

Параметры
объем текущий объем
Возврат
  • новый экземпляр XlaSparseCoreAdam

общедоступный вывод <Float> updateEmbeddingTable ()

публичный вывод <Float> updateMomentum ()

публичный вывод <Float> updateVelocity ()