org.tensorflow.op.train

Class SparseApplyMomentum.Options

java.lang.Object

org.tensorflow.op.train.SparseApplyMomentum.Options

Enclosing class:

SparseApplyMomentum<T>

public static class SparseApplyMomentum.Options
extends Object

Optional attributes for SparseApplyMomentum

Method Summary

Modifier and Type	Method and Description
SparseApplyMomentum.Options	useLocking(Boolean useLocking)
SparseApplyMomentum.Options	useNesterov(Boolean useNesterov)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

useLocking

public SparseApplyMomentum.Options useLocking(Boolean useLocking)

Parameters:

useLocking - 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.

useNesterov

public SparseApplyMomentum.Options useNesterov(Boolean useNesterov)

Parameters:

useNesterov - If `True`, the tensor passed to compute grad will be var - lr * momentum * accum, so in the end, the var you get is actually var - lr * momentum * accum.