| 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.sru(SDVariable x,
SDVariable initialC,
SDVariable mask,
SRUWeights SRUWeights)
The SRU layer.
|
SDVariable |
SDRNN.sru(SDVariable x,
SDVariable initialC,
SRUWeights SRUWeights)
The SRU layer.
|
SDVariable |
SDRNN.sru(String name,
SDVariable x,
SDVariable initialC,
SDVariable mask,
SRUWeights SRUWeights)
The SRU layer.
|
SDVariable |
SDRNN.sru(String name,
SDVariable x,
SDVariable initialC,
SRUWeights SRUWeights)
The SRU layer.
|
SDVariable |
SDRNN.sruCell(SDVariable x,
SDVariable cLast,
SRUWeights SRUWeights)
The SRU layer.
|
SDVariable |
SDRNN.sruCell(String name,
SDVariable x,
SDVariable cLast,
SRUWeights SRUWeights)
The SRU layer.
|
| Constructor and Description |
|---|
SRU(INDArray x,
INDArray initialC,
INDArray mask,
SRUWeights sruWeights) |
SRU(INDArray x,
INDArray initialC,
SRUWeights sruWeights) |
SRU(@NonNull SameDiff sameDiff,
@NonNull SDVariable x,
@NonNull SDVariable initialC,
SDVariable mask,
@NonNull SRUWeights weights) |
SRUCell(INDArray x,
INDArray cLast,
SRUWeights sruWeights) |
SRUCell(SameDiff sameDiff,
SDVariable x,
SDVariable cLast,
SRUWeights weights) |
| Modifier and Type | Method and Description |
|---|---|
INDArray |
NDRNN.sru(INDArray x,
INDArray initialC,
INDArray mask,
SRUWeights SRUWeights)
The SRU layer.
|
INDArray |
NDRNN.sru(INDArray x,
INDArray initialC,
SRUWeights SRUWeights)
The SRU layer.
|
INDArray |
NDRNN.sruCell(INDArray x,
INDArray cLast,
SRUWeights SRUWeights)
The SRU layer.
|
Copyright © 2021. All rights reserved.