org.deeplearning4j.nn.weights.embeddings

Class ArrayEmbeddingInitializer

java.lang.Object

org.deeplearning4j.nn.weights.embeddings.ArrayEmbeddingInitializer

All Implemented Interfaces:

Serializable, EmbeddingInitializer

public class ArrayEmbeddingInitializer
extends Object
implements EmbeddingInitializer

Embedding layer initialization from a specified array

Author:

Alex Black

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ArrayEmbeddingInitializer(INDArray embeddings)

Method Summary

Modifier and Type	Method and Description
boolean	jsonSerializable()
void	loadWeightsInto(INDArray array) Load the weights into the specified INDArray
int	vectorSize()
long	vocabSize()

Methods inherited from class java.lang.Object

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

Constructor Detail

ArrayEmbeddingInitializer

public ArrayEmbeddingInitializer(@NonNull
                                 INDArray embeddings)

Method Detail

loadWeightsInto

public void loadWeightsInto(INDArray array)

Description copied from interface: EmbeddingInitializer

Load the weights into the specified INDArray

Specified by:

loadWeightsInto in interface EmbeddingInitializer

Parameters:

array - Array of shape [vocabSize, vectorSize]

vocabSize

public long vocabSize()

Specified by:

vocabSize in interface EmbeddingInitializer

Returns:

Size of the vocabulary

vectorSize

public int vectorSize()

Specified by:

vectorSize in interface EmbeddingInitializer

Returns:

Size of each vector

jsonSerializable

public boolean jsonSerializable()

Specified by:

jsonSerializable in interface EmbeddingInitializer

Returns:

True if the embedding initializer can be safely serialized as JSON