Package ai.djl.nn.core
Contains classes that define simple neural network operations.
-
Interface Summary Interface Description AbstractEmbedding<T> An Embedding maps elements of type T to a 1-Dimensional representativeNDArray
s.AbstractIndexedEmbedding<T> AnAbstractEmbedding
where each embedded item can be assigned an integer index. -
Class Summary Class Description ConstantEmbedding AnAbstractIndexedEmbedding
that always returns a constant value.Embedding<T> An Embedding block map a collection of items to 1-Dimensional representativeNDArray
s.Embedding.BaseBuilder<T,B extends Embedding.BaseBuilder<T,B>> Linear A Linear block applies a linear transformation \(Y = XW^T + b\).Linear.Builder Prelu Applies Leaky Parametric ReLU activation element-wise to the input.