ApplyMomentum

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

Обновите '*var' в соответствии со схемой импульса.

Установите use_nesterov = True, если вы хотите использовать импульс Нестерова.

аккум = аккум * импульс + град вар -= lr * аккум

Вложенные классы

сорт ПрименитьMomentum.Параметры Дополнительные атрибуты для ApplyMomentum

Константы

Нить OP_NAME Название этой операции, известное основному движку TensorFlow.

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

Выход <Т>
какВывод ()
Возвращает символический дескриптор тензора.
статический <T расширяет TType > ApplyMomentum <T>
create ( Область действия, Операнд <T> var, Операнд <T> accum, Операнд <T> lr, Операнд <T> grad, Операнд <T> импульс, Опции... опции)
Фабричный метод для создания класса, обертывающего новую операцию ApplyMomentum.
Выход <Т>
вне ()
То же, что «вар».
статический ApplyMomentum.Options
useLocking (логическое значение useLocking)
статический ApplyMomentum.Options
useNesterov (логическое значение useNesterov)

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

Константы

общедоступная статическая финальная строка OP_NAME

Название этой операции, известное основному движку TensorFlow.

Постоянное значение: «ApplyMomentum».

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

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

Возвращает символический дескриптор тензора.

Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.

public static ApplyMomentum <T> create ( Область действия, Операнд <T> var, Операнд <T> accum, Операнд <T> lr, Операнд <T> град, Операнд <T> импульс, Параметры... параметры)

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

Параметры
объем текущий объем
вар Должно быть из переменной().
накапливать Должно быть из переменной().
лр Коэффициент масштабирования. Должно быть скаляр.
выпускник Градиент.
импульс Импульс. Должно быть скаляр.
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр ApplyMomentum

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

То же, что «вар».

общедоступный статический ApplyMomentum.Options useLocking (логическое значение useLocking)

Параметры
использоватьLocking Если `True`, обновление тензоров var и accum будет защищено блокировкой; в противном случае поведение не определено, но может вызывать меньше конфликтов.

public static ApplyMomentum.Options useNesterov (логическое значение useNesterov)

Параметры
использовать Нестерова Если `True`, тензор, передаваемый для вычисления градиента, будет иметь вид var - lr * импульс * аккум, поэтому в конечном итоге вы получите var - lr * импульс * аккум.