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>

The Builder to construct a LSTM type of Block.

Field Summary

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

activation, clipLstmState, dropRate, lstmStateClipMax, lstmStateClipMin, numStackedLayers, stateOutputs, stateSize, useBidirectional, useSequenceLength

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
LSTM	build() Builds a LSTM block.
LSTM.Builder	optLstmStateClipMin(float lstmStateClipMin, float lstmStateClipMax) Sets the minimum and maximum clip value of LSTM states.
protected LSTM.Builder	self()

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

optBidrectional, optDropRate, optStateOutput, setNumStackedLayers, setSequenceLength, setStateSize

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

self

protected LSTM.Builder self()

Specified by:

self in class RecurrentBlock.BaseBuilder<LSTM.Builder>

optLstmStateClipMin

public LSTM.Builder optLstmStateClipMin(float lstmStateClipMin,
                                        float lstmStateClipMax)

Sets the minimum and maximum clip value of LSTM states.

Parameters:

lstmStateClipMin - the minimum clip value of LSTM states

lstmStateClipMax - the maximum clip value of LSTM states

Returns:

this Builder

build

public LSTM build()

Builds a LSTM block.

Returns:

the LSTM block