tensorflow:: ops:: ResourceApplyKerasMomentum
#include <training_ops.h>
Update '*var' according to the momentum scheme.
Summary
Set use_nesterov = True if you want to use Nesterov momentum.
accum = accum * momentum - lr * grad var += accum
Args:
- scope: A Scope object
 - var: Should be from a Variable().
 - accum: Should be from a Variable().
 - lr: Scaling factor. Must be a scalar.
 - grad: The gradient.
 - momentum: Momentum. Must be a scalar.
 
Optional attributes (see Attrs):
- use_locking: If 
True, updating of the var and accum tensors will be protected by a lock; otherwise the behavior is undefined, but may exhibit less contention. - use_nesterov: If 
True, the tensor passed to compute grad will be var + momentum * accum, so in the end, the var you get is actually var + momentum * accum. 
Returns:
- the created 
Operation 
Constructors and Destructors | 
|
|---|---|
ResourceApplyKerasMomentum(const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum)
 | 
|
ResourceApplyKerasMomentum(const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum, const ResourceApplyKerasMomentum::Attrs & attrs)
 | 
Public attributes | 
|
|---|---|
operation
 | 
|
Public functions | 
|
|---|---|
operator::tensorflow::Operation() const 
 | 
 | 
Public static functions | 
|
|---|---|
UseLocking(bool x)
 | 
|
UseNesterov(bool x)
 | 
|
Structs | 
|
|---|---|
| 
tensorflow:: | 
 Optional attribute setters for ResourceApplyKerasMomentum.  | 
Public attributes
operation
Operation operation
Public functions
ResourceApplyKerasMomentum
ResourceApplyKerasMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum )
ResourceApplyKerasMomentum
ResourceApplyKerasMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum, const ResourceApplyKerasMomentum::Attrs & attrs )
operator::tensorflow::Operation
operator::tensorflow::Operation() const
Public static functions
UseLocking
Attrs UseLocking( bool x )
UseNesterov
Attrs UseNesterov( bool x )