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