tensorflow :: ops :: ResourceSparseApplyMomentum

Momentum şemasına göre '* var' ve '* biriktirme' içindeki ilgili girişleri güncelleyin.

Özet

Nesterov momentumunu kullanmak istiyorsanız use_nesterov = True ayarlayın.

Yani not verdiğimiz satırlar için var ve biriktirmeyi şu şekilde güncelliyoruz:

birikim = birikim * momentum + grad var - = lr * birikim

Argümanlar:

• kapsam: Bir Scope nesnesi
• var: Değişken () 'den olmalıdır.
• birikim: Bir Değişken () 'den olmalıdır.
• lr: Öğrenme oranı. Skaler olmalı.
• grad: Gradyan.
• indisler: var ve akümül'ün ilk boyutundaki indislerin bir vektörü.
• momentum: Momentum. Skaler olmalı.

İsteğe bağlı özellikler (bkz. Attrs ):

• use_locking: True ise, var ve akümülörlerin güncellenmesi bir kilit ile korunacaktır; aksi takdirde davranış tanımsızdır, ancak daha az çekişme gösterebilir.
• use_nesterov: True , grad'ı hesaplamak için aktarılan tensör var - lr * momentum * birikim olacaktır, bu yüzden sonunda elde ettiğiniz değişken aslında var - lr * momentum * biriktir.

İadeler:

• oluşturulan Operation

Genel özellikler

Operation operation

 ResourceSparseApplyMomentum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input accum,
  ::tensorflow::Input lr,
  ::tensorflow::Input grad,
  ::tensorflow::Input indices,
  ::tensorflow::Input momentum
)

 ResourceSparseApplyMomentum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input accum,
  ::tensorflow::Input lr,
  ::tensorflow::Input grad,
  ::tensorflow::Input indices,
  ::tensorflow::Input momentum,
  const ResourceSparseApplyMomentum::Attrs & attrs
)

 operator::tensorflow::Operation() const 

Attrs UseLocking(
  bool x
)

Attrs UseNesterov(
  bool x
)