Class ArrayEmbeddingInitializer
- java.lang.Object
-
- org.deeplearning4j.nn.weights.embeddings.ArrayEmbeddingInitializer
-
- All Implemented Interfaces:
Serializable
,EmbeddingInitializer
public class ArrayEmbeddingInitializer extends Object implements EmbeddingInitializer
- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description ArrayEmbeddingInitializer(@NonNull INDArray embeddings)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description boolean
jsonSerializable()
void
loadWeightsInto(INDArray array)
Load the weights into the specified INDArrayint
vectorSize()
long
vocabSize()
-
-
-
Constructor Detail
-
ArrayEmbeddingInitializer
public ArrayEmbeddingInitializer(@NonNull @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 interfaceEmbeddingInitializer
- Parameters:
array
- Array of shape [vocabSize, vectorSize]
-
vocabSize
public long vocabSize()
- Specified by:
vocabSize
in interfaceEmbeddingInitializer
- Returns:
- Size of the vocabulary
-
vectorSize
public int vectorSize()
- Specified by:
vectorSize
in interfaceEmbeddingInitializer
- Returns:
- Size of each vector
-
jsonSerializable
public boolean jsonSerializable()
- Specified by:
jsonSerializable
in interfaceEmbeddingInitializer
- Returns:
- True if the embedding initializer can be safely serialized as JSON
-
-