ResourceApplyKerasMomentum

publiczna klasa końcowa ResourceApplyKerasMomentum

Zaktualizuj „*var” zgodnie ze schematem pędu.

Ustaw use_nesterov = True, jeśli chcesz użyć pędu Niestierowa.

accum = accum * pęd - lr * grad var += accum

Klasy zagnieżdżone

klasa ResourceApplyKerasMomentum.Opcje Opcjonalne atrybuty ResourceApplyKerasMomentum

Metody publiczne

statyczny <T> ResourceApplyKerasMomentum
utwórz (Zakres zasięgu , Operand <?> var, Operand <?> accum, Operand <T> lr, Operand <T> grad, Operand <T> pęd, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ResourceApplyKerasMomentum.
statyczny ResourceApplyKerasMomentum.Options
useLocking (boolowski useLocking)
statyczny ResourceApplyKerasMomentum.Options
useNesterov (Boolean useNesterov)

Metody dziedziczone

Metody publiczne

public static ResourceApplyKerasMomentum create ( Zakres zakresu, Operand <?> var, Operand <?> accum, Operand <T> lr, Operand <T> grad, Operand <T> pęd, Opcje... opcje)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ResourceApplyKerasMomentum.

Parametry
zakres aktualny zakres
odm Powinno pochodzić ze zmiennej ().
gromadzić Powinno pochodzić ze zmiennej ().
lr Współczynnik skalowania. Musi być skalarem.
absolwent Gradient.
pęd Pęd. Musi być skalarem.
opcje przenosi opcjonalne wartości atrybutów
Zwroty
  • nowe wystąpienie ResourceApplyKerasMomentum

publiczny statyczny ResourceApplyKerasMomentum.Options useLocking (boolean useLocking)

Parametry
użyjBlokowanie Jeśli `True`, aktualizacja tensorów var i accum będzie zabezpieczona blokadą; w przeciwnym razie zachowanie jest niezdefiniowane, ale może wykazywać mniejszą rywalizację.

public static ResourceApplyKerasMomentum.Options useNesterov (Boolean useNesterov)

Parametry
użyj Niestierowa Jeśli „True”, tensor przekazany do obliczenia grad będzie miał postać var ​​+ pęd * accum, więc ostatecznie otrzymana wartość var ​​będzie wynosić w rzeczywistości var + pęd * accum.