ai.djl.basicmodelzoo.nlp

Class SimpleTextDecoder

java.lang.Object

ai.djl.nn.AbstractBlock

ai.djl.modality.nlp.Decoder

ai.djl.basicmodelzoo.nlp.SimpleTextDecoder

All Implemented Interfaces:

ai.djl.nn.Block

public class SimpleTextDecoder
extends ai.djl.modality.nlp.Decoder

SimpleTextDecoder implements a Decoder that employs a RecurrentBlock to decode text input.

Field Summary

Fields inherited from class ai.djl.modality.nlp.Decoder

block

Fields inherited from class ai.djl.nn.AbstractBlock

children, inputNames, inputShapes, parameters, version

Constructor Summary

Constructors

Constructor and Description
SimpleTextDecoder(ai.djl.nn.recurrent.RecurrentBlock recurrentBlock, int vocabSize) Contructs a new instance of SimpleTextDecoder with the given RecurrentBlock.
SimpleTextDecoder(ai.djl.modality.nlp.embedding.TrainableTextEmbedding trainableTextEmbedding, ai.djl.nn.recurrent.RecurrentBlock recurrentBlock, long vocabSize) Contructs a new instance of SimpleTextDecoder with the given RecurrentBlock.

Method Summary

Modifier and Type	Method and Description
protected ai.djl.ndarray.NDList	forwardInternal(ai.djl.training.ParameterStore parameterStore, ai.djl.ndarray.NDList inputs, boolean training, ai.djl.util.PairList<java.lang.String,java.lang.Object> params)

Methods inherited from class ai.djl.modality.nlp.Decoder

getOutputShapes, initializeChildBlocks, loadParameters, saveParameters

Methods inherited from class ai.djl.nn.AbstractBlock

addChildBlock, addParameter, beforeInitialize, cast, clear, describeInput, forward, forward, forwardInternal, getChildren, getDirectParameters, getParameters, initialize, isInitialized, loadMetadata, prepare, readInputShapes, saveInputShapes, saveMetadata, setInitializer, setInitializer, setInitializer, toString

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.nn.Block

forward, validateLayout

Constructor Detail

SimpleTextDecoder

public SimpleTextDecoder(ai.djl.nn.recurrent.RecurrentBlock recurrentBlock,
                         int vocabSize)

Contructs a new instance of SimpleTextDecoder with the given RecurrentBlock. Use this constructor if you are planning to use pre-trained embeddings that don't need further training.

Parameters:

recurrentBlock - the recurrent block to be used to decode

vocabSize - the size of the Vocabulary

SimpleTextDecoder

public SimpleTextDecoder(ai.djl.modality.nlp.embedding.TrainableTextEmbedding trainableTextEmbedding,
                         ai.djl.nn.recurrent.RecurrentBlock recurrentBlock,
                         long vocabSize)

Contructs a new instance of SimpleTextDecoder with the given RecurrentBlock. Use this constructor if you are planning to use pre-trained embeddings that don't need further training.

Parameters:

trainableTextEmbedding - the TrainableTextEmbedding to train embeddings with

recurrentBlock - the recurrent block to be used to decode

vocabSize - the size of the Vocabulary

Method Detail

forwardInternal

protected ai.djl.ndarray.NDList forwardInternal(ai.djl.training.ParameterStore parameterStore,
                                                ai.djl.ndarray.NDList inputs,
                                                boolean training,
                                                ai.djl.util.PairList<java.lang.String,java.lang.Object> params)

Overrides:

forwardInternal in class ai.djl.modality.nlp.Decoder