| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.train |
| Modifier and Type | Method and Description |
|---|---|
<T> ApplyAdam<T> |
TrainOps.applyAdam(Operand<T> var,
Operand<T> m,
Operand<T> v,
Operand<T> beta1Power,
Operand<T> beta2Power,
Operand<T> lr,
Operand<T> beta1,
Operand<T> beta2,
Operand<T> epsilon,
Operand<T> grad,
ApplyAdam.Options... options)
Builds an
ApplyAdam operation |
| Modifier and Type | Method and Description |
|---|---|
static ApplyAdam.Options |
ApplyAdam.useLocking(Boolean useLocking) |
ApplyAdam.Options |
ApplyAdam.Options.useLocking(Boolean useLocking) |
static ApplyAdam.Options |
ApplyAdam.useNesterov(Boolean useNesterov) |
ApplyAdam.Options |
ApplyAdam.Options.useNesterov(Boolean useNesterov) |
| Modifier and Type | Method and Description |
|---|---|
static <T> ApplyAdam<T> |
ApplyAdam.create(Scope scope,
Operand<T> var,
Operand<T> m,
Operand<T> v,
Operand<T> beta1Power,
Operand<T> beta2Power,
Operand<T> lr,
Operand<T> beta1,
Operand<T> beta2,
Operand<T> epsilon,
Operand<T> grad,
ApplyAdam.Options... options)
Factory method to create a class wrapping a new ApplyAdam operation.
|
Copyright © 2015–2019. All rights reserved.