| 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.lstmblock(SDVariable x,
LSTMWeights LSTMWeights,
LSTMConfiguration LSTMConfiguration)
The LSTM block
|
SDVariable |
SDRNN.lstmblock(SDVariable maxTSLength,
SDVariable x,
SDVariable cLast,
SDVariable yLast,
LSTMWeights LSTMWeights,
LSTMConfiguration LSTMConfiguration)
The LSTM block
|
SDVariable |
SDRNN.lstmblock(String name,
SDVariable x,
LSTMWeights LSTMWeights,
LSTMConfiguration LSTMConfiguration)
The LSTM block
|
SDVariable |
SDRNN.lstmblock(String name,
SDVariable maxTSLength,
SDVariable x,
SDVariable cLast,
SDVariable yLast,
LSTMWeights LSTMWeights,
LSTMConfiguration LSTMConfiguration)
The LSTM block
|
SDVariable[] |
SDRNN.lstmCell(SDVariable x,
SDVariable cLast,
SDVariable yLast,
LSTMWeights LSTMWeights,
LSTMConfiguration LSTMConfiguration)
The LSTM cell.
|
SDVariable[] |
SDRNN.lstmCell(String[] names,
SDVariable x,
SDVariable cLast,
SDVariable yLast,
LSTMWeights LSTMWeights,
LSTMConfiguration LSTMConfiguration)
The LSTM cell.
|
| Constructor and Description |
|---|
LSTMBlock(INDArray x,
INDArray cLast,
INDArray yLast,
INDArray maxTSLength,
LSTMWeights lstmWeights,
LSTMConfiguration lstmConfiguration) |
LSTMBlock(@NonNull SameDiff sameDiff,
SDVariable maxTSLength,
SDVariable x,
SDVariable cLast,
SDVariable yLast,
LSTMWeights weights,
LSTMConfiguration configuration) |
LSTMBlockCell(INDArray x,
INDArray cLast,
INDArray yLast,
LSTMWeights lstmWeights,
LSTMConfiguration lstmConfiguration) |
LSTMBlockCell(SameDiff sameDiff,
SDVariable x,
SDVariable cLast,
SDVariable yLast,
LSTMWeights weights,
LSTMConfiguration configuration) |
| Modifier and Type | Method and Description |
|---|---|
INDArray |
NDRNN.lstmblock(INDArray maxTSLength,
INDArray x,
INDArray cLast,
INDArray yLast,
LSTMWeights LSTMWeights,
LSTMConfiguration LSTMConfiguration)
The LSTM block
|
INDArray |
NDRNN.lstmblock(INDArray x,
LSTMWeights LSTMWeights,
LSTMConfiguration LSTMConfiguration)
The LSTM block
|
INDArray[] |
NDRNN.lstmCell(INDArray x,
INDArray cLast,
INDArray yLast,
LSTMWeights LSTMWeights,
LSTMConfiguration LSTMConfiguration)
The LSTM cell.
|
Copyright © 2021. All rights reserved.