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 RNN
build()
Builds aRNN
block.protected RNN.Builder
self()
RNN.Builder
setActivation(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:
self
in 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
-
-