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"