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