Package | Description |
---|---|
org.nd4j.autodiff.samediff.ops | |
org.nd4j.linalg.api.ops.impl.layers.recurrent |
Modifier and Type | Method and Description |
---|---|
SRULayerOutputs |
SDRNN.sru(SDVariable x,
SDVariable initialC,
SDVariable mask,
SRUWeights weights)
|
SRULayerOutputs |
SDRNN.sru(SDVariable x,
SDVariable initialC,
SRUWeights weights)
|
SRULayerOutputs |
SDRNN.sru(String baseName,
SDVariable x,
SDVariable initialC,
SDVariable mask,
SRUWeights weights)
The SRU layer.
|
SRULayerOutputs |
SDRNN.sru(String baseName,
SDVariable x,
SDVariable initialC,
SRUWeights weights)
|
SRUCellOutputs |
SDRNN.sruCell(SDVariable x,
SDVariable cLast,
SRUWeights weights)
|
SRUCellOutputs |
SDRNN.sruCell(String baseName,
SDVariable x,
SDVariable cLast,
SRUWeights weights)
The SRU cell.
|
Constructor and Description |
---|
SRU(SameDiff sameDiff,
SDVariable x,
SDVariable initialC,
SDVariable mask,
SRUWeights weights) |
SRUCell(SameDiff sameDiff,
SDVariable x,
SDVariable cLast,
SRUWeights weights) |
Copyright © 2019. All rights reserved.