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