Package ai.djl.nn.recurrent
Class GRU.Builder
- java.lang.Object
-
- ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder<GRU.Builder>
-
- ai.djl.nn.recurrent.GRU.Builder
-
- Enclosing class:
- GRU
public static final class GRU.Builder extends RecurrentBlock.BaseBuilder<GRU.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 GRUbuild()Builds aGRUblock.protected GRU.Builderself()-
Methods inherited from class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
optBatchFirst, optBidirectional, optDropRate, optHasBiases, optReturnState, setNumLayers, setStateSize
-
-
-
-
Method Detail
-
self
protected GRU.Builder self()
- Specified by:
selfin classRecurrentBlock.BaseBuilder<GRU.Builder>
-
-