Package ai.djl.nn.core

Contains classes that define simple neural network operations.

See: Description

Interface Summary

Interface	Description
AbstractEmbedding<T>	An Embedding maps elements of type T to a 1-Dimensional representative NDArrays.
AbstractIndexedEmbedding<T>	An AbstractEmbedding where each embedded item can be assigned an integer index.

Class Summary

Class	Description
ConstantEmbedding	An AbstractIndexedEmbedding that always returns a constant value.
Embedding<T>	An Embedding block map a collection of items to 1-Dimensional representative NDArrays.
Embedding.BaseBuilder<T,B extends Embedding.BaseBuilder<T,B>>	The Builder to construct a Embedding type of Block.
Linear	A Linear block applies a linear transformation \(Y = XW^T + b\).
Linear.Builder	The Builder to construct a Linear type of Block.
Prelu	Applies Leaky Parametric ReLU activation element-wise to the input.

Package ai.djl.nn.core Description

Contains classes that define simple neural network operations.