| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff.ops | |
| org.nd4j.linalg.api.ops.impl.layers.recurrent | |
| org.nd4j.linalg.factory.ops |
| Modifier and Type | Method and Description |
|---|---|
SDVariable[] |
SDRNN.gruCell(SDVariable x,
SDVariable hLast,
GRUWeights GRUWeights)
The GRU cell.
|
SDVariable[] |
SDRNN.gruCell(String[] names,
SDVariable x,
SDVariable hLast,
GRUWeights GRUWeights)
The GRU cell.
|
| Constructor and Description |
|---|
GRUCell(INDArray x,
INDArray hLast,
GRUWeights gruWeights) |
GRUCell(SameDiff sameDiff,
SDVariable x,
SDVariable hLast,
GRUWeights weights) |
| Modifier and Type | Method and Description |
|---|---|
INDArray[] |
NDRNN.gruCell(INDArray x,
INDArray hLast,
GRUWeights GRUWeights)
The GRU cell.
|
Copyright © 2021. All rights reserved.