| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.train |
| Modifier and Type | Method and Description |
|---|---|
<T> ApplyRmsProp<T> |
TrainOps.applyRmsProp(Operand<T> var,
Operand<T> ms,
Operand<T> mom,
Operand<T> lr,
Operand<T> rho,
Operand<T> momentum,
Operand<T> epsilon,
Operand<T> grad,
ApplyRmsProp.Options... options)
Builds an
ApplyRmsProp operation |
| Modifier and Type | Method and Description |
|---|---|
static ApplyRmsProp.Options |
ApplyRmsProp.useLocking(Boolean useLocking) |
ApplyRmsProp.Options |
ApplyRmsProp.Options.useLocking(Boolean useLocking) |
| Modifier and Type | Method and Description |
|---|---|
static <T> ApplyRmsProp<T> |
ApplyRmsProp.create(Scope scope,
Operand<T> var,
Operand<T> ms,
Operand<T> mom,
Operand<T> lr,
Operand<T> rho,
Operand<T> momentum,
Operand<T> epsilon,
Operand<T> grad,
ApplyRmsProp.Options... options)
Factory method to create a class wrapping a new ApplyRmsProp operation.
|
Copyright © 2015–2019. All rights reserved.