Package ai.djl.nn.core

Contains classes that define simple neural network operations.

See: Description

Class Summary

Class	Description
Embedding<T>	An Embedding block map a collection of items to 1-Dimensional representative NDArrays.
Embedding.Builder<T>	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.