Package ai.djl.nn.recurrent
Class LSTM.Builder
- java.lang.Object
-
- ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder<LSTM.Builder>
-
- ai.djl.nn.recurrent.LSTM.Builder
-
- Enclosing class:
- LSTM
public static final class LSTM.Builder extends RecurrentBlock.BaseBuilder<LSTM.Builder>
-
-
Field Summary
-
Fields inherited from class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
activation, batchFirst, bidirectional, dropRate, hasBiases, numLayers, returnState, stateSize
-
-
Constructor Summary
Constructors Constructor Description Builder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description LSTMbuild()Builds aLSTMblock.protected LSTM.Builderself()-
Methods inherited from class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
optBatchFirst, optBidirectional, optDropRate, optHasBiases, optReturnState, setNumLayers, setStateSize
-
-
-
-
Method Detail
-
self
protected LSTM.Builder self()
- Specified by:
selfin classRecurrentBlock.BaseBuilder<LSTM.Builder>
-
-