Package | Description |
---|---|
org.tensorflow.op | |
org.tensorflow.op.train |
Modifier and Type | Method and Description |
---|---|
<T,U extends Number> |
TrainOps.sparseApplyRmsProp(Operand<T> var,
Operand<T> ms,
Operand<T> mom,
Operand<T> lr,
Operand<T> rho,
Operand<T> momentum,
Operand<T> epsilon,
Operand<T> grad,
Operand<U> indices,
SparseApplyRmsProp.Options... options)
Builds an
SparseApplyRmsProp operation |
<T,U extends Number> |
TrainOps.sparseApplyRmsProp(Operand<T> var,
Operand<T> ms,
Operand<T> mom,
Operand<T> lr,
Operand<T> rho,
Operand<T> momentum,
Operand<T> epsilon,
Operand<T> grad,
Operand<U> indices,
SparseApplyRmsProp.Options... options)
Builds an
SparseApplyRmsProp operation |
Modifier and Type | Method and Description |
---|---|
static <T,U extends Number> |
SparseApplyRmsProp.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,
Operand<U> indices,
SparseApplyRmsProp.Options... options)
Factory method to create a class wrapping a new SparseApplyRmsProp operation.
|
static <T,U extends Number> |
SparseApplyRmsProp.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,
Operand<U> indices,
SparseApplyRmsProp.Options... options)
Factory method to create a class wrapping a new SparseApplyRmsProp operation.
|
Copyright © 2022. All rights reserved.