Package | Description |
---|---|
org.tensorflow.op | |
org.tensorflow.op.train |
Modifier and Type | Method and Description |
---|---|
<T> ApplyMomentum<T> |
TrainOps.applyMomentum(Operand<T> var,
Operand<T> accum,
Operand<T> lr,
Operand<T> grad,
Operand<T> momentum,
ApplyMomentum.Options... options)
Builds an
ApplyMomentum operation |
<T> ApplyMomentum<T> |
TrainOps.applyMomentum(Operand<T> var,
Operand<T> accum,
Operand<T> lr,
Operand<T> grad,
Operand<T> momentum,
ApplyMomentum.Options... options)
Builds an
ApplyMomentum operation |
Modifier and Type | Method and Description |
---|---|
static ApplyMomentum.Options |
ApplyMomentum.useLocking(Boolean useLocking) |
ApplyMomentum.Options |
ApplyMomentum.Options.useLocking(Boolean useLocking) |
static ApplyMomentum.Options |
ApplyMomentum.useLocking(Boolean useLocking) |
ApplyMomentum.Options |
ApplyMomentum.Options.useLocking(Boolean useLocking) |
static ApplyMomentum.Options |
ApplyMomentum.useNesterov(Boolean useNesterov) |
ApplyMomentum.Options |
ApplyMomentum.Options.useNesterov(Boolean useNesterov) |
static ApplyMomentum.Options |
ApplyMomentum.useNesterov(Boolean useNesterov) |
ApplyMomentum.Options |
ApplyMomentum.Options.useNesterov(Boolean useNesterov) |
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 wrapping a new ApplyMomentum operation.
|
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 wrapping a new ApplyMomentum operation.
|
Copyright © 2022. All rights reserved.