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>

The Builder to construct a GRU type of Block.

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

Modifier and Type	Method	Description
GRU	build()	Builds a GRU block.
protected GRU.Builder	self()

Methods inherited from class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder

optBatchFirst, optBidirectional, optDropRate, optHasBiases, optReturnState, setNumLayers, setStateSize

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

Builder

public Builder()

Method Details

self

protected GRU.Builder self()

Specified by:

self in class RecurrentBlock.BaseBuilder<GRU.Builder>

build

public GRU build()

Builds a GRU block.

Returns:

the GRU block