| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.core |
| Modifier and Type | Method and Description |
|---|---|
<T> ApplyMomentum<T> |
Ops.applyMomentum(Operand<T> var,
Operand<T> accum,
Operand<T> lr,
Operand<T> grad,
Operand<T> momentum,
ApplyMomentum.Options... options)
Adds an
ApplyMomentum operation to the graph |
| Modifier and Type | Method and Description |
|---|---|
static <T> ApplyMomentum<T> |
ApplyMomentum.create(Scope scope,
Operand<T> var,
Operand<T> accum,
Operand<T> lr,
Operand<T> grad,
Operand<T> momentum,
ApplyMomentum.Options... options)
Factory method to create a class to wrap a new ApplyMomentum operation to the graph.
|
Copyright © 2015–2019. All rights reserved.