tf.train.MomentumOptimizer(
learning_rate,
momentum,
use_locking=False,
use_nesterov=False,
name='Momentum'
)
參數:
learning_rate: (學習率)張量或者浮點數
momentum: (動量)張量或者浮點數
use_locking: 為True時鎖定更新
use_nesterov: 為True時,使用 Nesterov Momentum
name: 梯度下降名稱,默認為 "Momentum"