ai.djl.nn.core

Class ConstantEmbedding

java.lang.Object

ai.djl.nn.AbstractBlock

ai.djl.nn.core.ConstantEmbedding

All Implemented Interfaces:

Block, AbstractEmbedding, AbstractIndexedEmbedding

public class ConstantEmbedding
extends AbstractBlock
implements AbstractIndexedEmbedding

An AbstractIndexedEmbedding that always returns a constant value.

Field Summary

Fields

Modifier and Type	Field and Description
protected NDArray	embedding

Fields inherited from class ai.djl.nn.AbstractBlock

children, inputNames, inputShapes, parameters, version

Constructor Summary

Constructors

Constructor and Description
ConstantEmbedding(NDArray embedding) Constructs a constant embedding with the given constant.

Method Summary

Modifier and Type	Method and Description
java.lang.Object	decode(byte[] byteArray) Decodes the given byte array into an object of input parameter type.
NDArray	embed(NDManager manager, java.lang.Object[] items) Embeds an array of items.
long	embed(java.lang.Object item) Embeds an item.
byte[]	encode(java.lang.Object input) Encodes an object of input type into a byte array.
protected NDList	forwardInternal(ParameterStore parameterStore, NDList inputs, boolean training, ai.djl.util.PairList<java.lang.String,java.lang.Object> params) A helper for Block.forward(ParameterStore, NDList, boolean, PairList) after initialization.
Shape[]	getOutputShapes(Shape[] inputShapes) Returns the expected output shapes of the block for the specified input shapes.
boolean	hasItem(java.lang.Object item) Returns whether an item is in the embedding.
void	loadParameters(NDManager manager, java.io.DataInputStream is) Loads the parameters from the given input stream.
void	saveParameters(java.io.DataOutputStream os) Writes the parameters of the block to the given outputStream.
java.util.Optional<?>	unembed(long index) Returns the item corresponding to the given index.

Methods inherited from class ai.djl.nn.AbstractBlock

addChildBlock, addParameter, beforeInitialize, cast, clear, describeInput, forward, forward, forwardInternal, getChildren, getDirectParameters, getParameters, initialize, initializeChildBlocks, 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

Field Detail

embedding

protected NDArray embedding

Constructor Detail

ConstantEmbedding

public ConstantEmbedding(NDArray embedding)

Constructs a constant embedding with the given constant.

Parameters:

embedding - the value to return for all embeddings

Method Detail

forwardInternal

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

A helper for Block.forward(ParameterStore, NDList, boolean, PairList) after initialization.

Specified by:

forwardInternal in class AbstractBlock

Parameters:

parameterStore - the parameter store

inputs - the input NDList

training - true for a training forward pass

params - optional parameters

Returns:

the output of the forward pass

getOutputShapes

public Shape[] getOutputShapes(Shape[] inputShapes)

Returns the expected output shapes of the block for the specified input shapes.

Specified by:

getOutputShapes in interface Block

Parameters:

inputShapes - the shapes of the inputs

Returns:

the expected output shapes of the block

saveParameters

public void saveParameters(java.io.DataOutputStream os)

Writes the parameters of the block to the given outputStream.

Specified by:

saveParameters in interface Block

Overrides:

saveParameters in class AbstractBlock

Parameters:

os - the outputstream to save the parameters to

loadParameters

public void loadParameters(NDManager manager,
                           java.io.DataInputStream is)

Loads the parameters from the given input stream.

Specified by:

loadParameters in interface Block

Overrides:

loadParameters in class AbstractBlock

Parameters:

manager - an NDManager to create the parameter arrays

is - the inputstream that stream the parameter values

unembed

public java.util.Optional<?> unembed(long index)

Returns the item corresponding to the given index.

Specified by:

unembed in interface AbstractIndexedEmbedding

Parameters:

index - the index

Returns:

the item corresponding to the given index

encode

public byte[] encode(java.lang.Object input)

Encodes an object of input type into a byte array. This is used in saving and loading the Embedding objects.

Specified by:

encode in interface AbstractIndexedEmbedding

Parameters:

input - the input object to be encoded

Returns:

the encoded byte array.

decode

public java.lang.Object decode(byte[] byteArray)

Decodes the given byte array into an object of input parameter type.

Specified by:

decode in interface AbstractIndexedEmbedding

Parameters:

byteArray - the byte array to be decoded

Returns:

the decode object of input parameter type

embed

public long embed(java.lang.Object item)

Embeds an item.

Specified by:

embed in interface AbstractIndexedEmbedding

Parameters:

item - the item to embed

Returns:

the index of the item in the embedding

embed

public NDArray embed(NDManager manager,
                     java.lang.Object[] items)

Embeds an array of items.

Specified by:

embed in interface AbstractEmbedding

Parameters:

manager - the manager for the new embeddings

items - the items to embed

Returns:

the embedding NDArray of Shape(items.length, embeddingSize)

hasItem

public boolean hasItem(java.lang.Object item)

Returns whether an item is in the embedding.

Specified by:

hasItem in interface AbstractEmbedding

Parameters:

item - the item to test

Returns:

true if the item is in the embedding