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