Package ai.djl.nn.recurrent
Class RNN.Builder
- java.lang.Object
-
- ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder<RNN.Builder>
-
- ai.djl.nn.recurrent.RNN.Builder
-
- Enclosing class:
- RNN
public static final class RNN.Builder extends RecurrentBlock.BaseBuilder<RNN.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 RNNbuild()Builds aRNNblock.protected RNN.Builderself()RNN.BuildersetActivation(RNN.Activation activation)Sets the activation for the RNN - ReLu or Tanh.-
Methods inherited from class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
optBatchFirst, optBidirectional, optDropRate, optHasBiases, optReturnState, setNumLayers, setStateSize
-
-
-
-
Method Detail
-
self
protected RNN.Builder self()
- Specified by:
selfin classRecurrentBlock.BaseBuilder<RNN.Builder>
-
setActivation
public RNN.Builder setActivation(RNN.Activation activation)
Sets the activation for the RNN - ReLu or Tanh.- Parameters:
activation- the activation- Returns:
- this Builder
-
-