Обновите '*var' в соответствии со схемой импульса.
Установите use_nesterov = True, если вы хотите использовать импульс Нестерова.
аккум = аккум * импульс + град вар -= lr * аккум
Вложенные классы
сорт | ПрименитьMomentum.Параметры | Дополнительные атрибуты для ApplyMomentum |
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
статический <T расширяет TType > ApplyMomentum <T> | |
Выход <Т> | вне () То же, что «вар». |
статический ApplyMomentum.Options | useLocking (логическое значение useLocking) |
статический ApplyMomentum.Options | useNesterov (логическое значение useNesterov) |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static ApplyMomentum <T> create ( Область действия, Операнд <T> var, Операнд <T> accum, Операнд <T> lr, Операнд <T> град, Операнд <T> импульс, Параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию ApplyMomentum.
Параметры
объем | текущий объем |
---|---|
вар | Должно быть из переменной(). |
накапливать | Должно быть из переменной(). |
лр | Коэффициент масштабирования. Должно быть скаляр. |
выпускник | Градиент. |
импульс | Импульс. Должно быть скаляр. |
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр ApplyMomentum
общедоступный статический ApplyMomentum.Options useLocking (логическое значение useLocking)
Параметры
использоватьLocking | Если `True`, обновление тензоров var и accum будет защищено блокировкой; в противном случае поведение не определено, но может вызывать меньше конфликтов. |
---|
public static ApplyMomentum.Options useNesterov (логическое значение useNesterov)
Параметры
использовать Нестерова | Если `True`, тензор, передаваемый для вычисления градиента, будет иметь вид var - lr * импульс * аккум, поэтому в конечном итоге вы получите var - lr * импульс * аккум. |
---|