ResourceSparseApplyMomentum

publiczna klasa końcowa ResourceSparseApplyMomentum

Zaktualizuj odpowiednie wpisy w „*var” i „*accum” zgodnie ze schematem pędu.

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

Oznacza to, że dla wierszy, dla których mamy grad, aktualizujemy var i accum w następujący sposób:

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

Klasy zagnieżdżone

klasa ResourceSparseApplyMomentum.Opcje Opcjonalne atrybuty ResourceSparseApplyMomentum

Stałe

Strunowy OP_NAME Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Metody publiczne

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

Metody dziedziczone

Stałe

publiczny statyczny końcowy ciąg znaków OP_NAME

Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Wartość stała: „ResourceSparseApplyMomentum”

Metody publiczne

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

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

Parametry
zakres aktualny zakres
odm Powinno pochodzić ze zmiennej ().
gromadzić Powinno pochodzić ze zmiennej ().
lr Szybkość uczenia się. Musi być skalarem.
absolwent Gradient.
indeksy Wektor indeksów do pierwszego wymiaru var i accum.
pęd Pęd. Musi być skalarem.
opcje przenosi opcjonalne wartości atrybutów
Zwroty
  • nową instancję ResourceSparseApplyMomentum

public static ResourceSparseApplyMomentum.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 ResourceSparseApplyMomentum.Options useNesterov (Boolean useNesterov)

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