Skip navigation links
A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 

A

abs() - Method in interface ai.djl.ndarray.NDArray
Returns the absolute value of this NDArray element-wise.
abs() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the absolute value of this NDArray element-wise.
AbstractAccuracy - Class in ai.djl.training.evaluator
Accuracy is an Evaluator that computes the accuracy score.
AbstractAccuracy(String, int) - Constructor for class ai.djl.training.evaluator.AbstractAccuracy
Creates an accuracy evaluator that computes accuracy across axis 1 along given index.
AbstractAccuracy(String, int, int) - Constructor for class ai.djl.training.evaluator.AbstractAccuracy
Creates an accuracy evaluator.
AbstractBlock - Class in ai.djl.nn
AbstractBlock is an abstract implementation of Block.
AbstractBlock() - Constructor for class ai.djl.nn.AbstractBlock
Constructs a new AbstractBlock instance.
AbstractBlock(byte) - Constructor for class ai.djl.nn.AbstractBlock
Builds an empty block with the given version for parameter serialization.
AbstractCompositeLoss - Class in ai.djl.training.loss
AbstractCompositeLoss is a Loss class that can combine other Losses together to make a larger loss.
AbstractCompositeLoss(String) - Constructor for class ai.djl.training.loss.AbstractCompositeLoss
Constructs a composite loss with the given name.
AbstractEmbedding<T> - Interface in ai.djl.nn.core
An Embedding maps elements of type T to a 1-Dimensional representative NDArrays.
AbstractIndexedEmbedding<T> - Interface in ai.djl.nn.core
An AbstractEmbedding where each embedded item can be assigned an integer index.
AbstractRepository - Class in ai.djl.repository
The AbstractRepository is the shared base for implementers of the Repository interface.
AbstractRepository(String, URI) - Constructor for class ai.djl.repository.AbstractRepository
 
AbstractSymbolBlock - Class in ai.djl.nn
AbstractSymbolBlock is an abstract implementation of SymbolBlock.
AbstractSymbolBlock() - Constructor for class ai.djl.nn.AbstractSymbolBlock
Constructs a new AbstractSymbolBlock instance.
AbstractSymbolBlock(byte) - Constructor for class ai.djl.nn.AbstractSymbolBlock
Builds an empty block with the given version for parameter serialization.
accuracy(NDList, NDList) - Method in class ai.djl.nn.transformer.BertMaskedLanguageModelLoss
Calculates the percentage of correctly predicted masked tokens.
accuracy(NDList, NDList) - Method in class ai.djl.nn.transformer.BertNextSentenceLoss
Calculates the fraction of correct predictions.
Accuracy - Class in ai.djl.training.evaluator
Accuracy is the AbstractAccuracy with multiple classes.
Accuracy() - Constructor for class ai.djl.training.evaluator.Accuracy
Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along the 0th index.
Accuracy(String, int) - Constructor for class ai.djl.training.evaluator.Accuracy
Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along given index.
Accuracy(String, int, int) - Constructor for class ai.djl.training.evaluator.Accuracy
Creates a multiclass accuracy evaluator.
accuracyHelper(NDList, NDList) - Method in class ai.djl.training.evaluator.AbstractAccuracy
A helper for classes extending AbstractAccuracy.
accuracyHelper(NDList, NDList) - Method in class ai.djl.training.evaluator.Accuracy
A helper for classes extending AbstractAccuracy.
accuracyHelper(NDList, NDList) - Method in class ai.djl.training.evaluator.BinaryAccuracy
A helper for classes extending AbstractAccuracy.
accuracyHelper(NDList, NDList) - Method in class ai.djl.training.evaluator.SingleShotDetectionAccuracy
A helper for classes extending AbstractAccuracy.
accuracyHelper(NDList, NDList) - Method in class ai.djl.training.evaluator.TopKAccuracy
A helper for classes extending AbstractAccuracy.
acos() - Method in interface ai.djl.ndarray.NDArray
Returns the inverse trigonometric cosine of this NDArray element-wise.
acos() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the inverse trigonometric cosine of this NDArray element-wise.
acosh() - Method in interface ai.djl.ndarray.NDArray
Returns the inverse hyperbolic cosine of this NDArray element-wise.
acosh() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the inverse hyperbolic cosine of this NDArray element-wise.
ACTION_RECOGNITION - Static variable in interface ai.djl.Application.CV
An application that accepts an image or video and classifies the action being done in it.
ActionSpace - Class in ai.djl.modality.rl
Contains the available actions that can be taken in an RlEnv.
ActionSpace() - Constructor for class ai.djl.modality.rl.ActionSpace
 
Activation - Class in ai.djl.nn
Utility class that provides activation functions and blocks.
activation - Variable in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
 
Adadelta - Class in ai.djl.training.optimizer
Adadelta is an Adadelta Optimizer.
Adadelta(Adadelta.Builder) - Constructor for class ai.djl.training.optimizer.Adadelta
Creates a new instance of Adadelta.
adadelta() - Static method in class ai.djl.training.optimizer.Optimizer
Returns a new instance of Adadelta.Builder that can build an Adadelta optimizer.
Adadelta.Builder - Class in ai.djl.training.optimizer
The Builder to construct an Adadelta object.
Adagrad - Class in ai.djl.training.optimizer
Adagrad is an AdaGrad Optimizer.
Adagrad(Adagrad.Builder) - Constructor for class ai.djl.training.optimizer.Adagrad
Creates a new instance of Adam optimizer.
adagrad() - Static method in class ai.djl.training.optimizer.Optimizer
Returns a new instance of Adagrad.Builder that can build an Adagrad optimizer.
Adagrad.Builder - Class in ai.djl.training.optimizer
The Builder to construct an Adagrad object.
Adam - Class in ai.djl.training.optimizer
Adam is a generalization of the AdaGrad Optimizer.
Adam(Adam.Builder) - Constructor for class ai.djl.training.optimizer.Adam
Creates a new instance of Adam optimizer.
adam() - Static method in class ai.djl.training.optimizer.Optimizer
Returns a new instance of Adam.Builder that can build an Adam optimizer.
Adam.Builder - Class in ai.djl.training.optimizer
The Builder to construct an Adam object.
add(byte[]) - Method in class ai.djl.modality.Input
Appends an item at the end of the input.
add(String) - Method in class ai.djl.modality.Input
Appends an item at the end of the input.
add(BytesSupplier) - Method in class ai.djl.modality.Input
Appends an item at the end of the input.
add(String, byte[]) - Method in class ai.djl.modality.Input
Adds a key/value pair to the input content.
add(String, String) - Method in class ai.djl.modality.Input
Adds a key/value pair to the input content.
add(String, BytesSupplier) - Method in class ai.djl.modality.Input
Adds a key/value pair to the input content.
add(int, String, BytesSupplier) - Method in class ai.djl.modality.Input
Inserts the specified element at the specified position in the input.
add(List<String>) - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Adds the given sentence to the DefaultVocabulary.
add(Number) - Method in interface ai.djl.ndarray.NDArray
Adds a number to this NDArray element-wise.
add(NDArray) - Method in interface ai.djl.ndarray.NDArray
Adds other NDArrays to this NDArray element-wise.
add(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Adds a number to this NDArray element-wise.
add(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Adds other NDArrays to this NDArray element-wise.
add(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Adds a number to the NDArray element-wise.
add(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Adds a NDArray to a number element-wise.
add(NDArray...) - Static method in class ai.djl.ndarray.NDArrays
Adds a NDArray to a NDArray element-wise.
add(long...) - Method in class ai.djl.ndarray.types.Shape
Joins this shape with axes.
add(Block) - Method in class ai.djl.nn.ParallelBlock
Adds the given Block to the block, which is one parallel branch.
add(Function<NDList, NDList>) - Method in class ai.djl.nn.ParallelBlock
Adds a LambdaBlock, that applies the given function, to the list of parallel branches.
add(Block) - Method in class ai.djl.nn.SequentialBlock
Adds the given Block to the block to be executed in order.
add(Function<NDList, NDList>) - Method in class ai.djl.nn.SequentialBlock
Adds a LambdaBlock that applies the given function to the sequence of blocks.
add(Hyperparameter<?>) - Method in class ai.djl.training.hyperparameter.param.HpSet
Adds a hyperparameter to the set.
add(Transform) - Method in class ai.djl.translate.Pipeline
Adds the given Transform to the list of transforms to be applied on the input when the transform method is called on this object.
add(int, Transform) - Method in class ai.djl.translate.Pipeline
Adds the given Transform to the list of transforms to be applied on the NDArray at the given index in the input NDList.
add(String, Transform) - Method in class ai.djl.translate.Pipeline
Adds the given Transform to the list of transforms to be applied on the NDArray with the given key as name in the input NDList.
addAccumulator(String) - Method in class ai.djl.training.evaluator.AbstractAccuracy
Adds an accumulator for the results of the evaluation with the given key.
addAccumulator(String) - Method in class ai.djl.training.evaluator.BoundingBoxError
Adds an accumulator for the results of the evaluation with the given key.
addAccumulator(String) - Method in class ai.djl.training.evaluator.Evaluator
Adds an accumulator for the results of the evaluation with the given key.
addAccumulator(String) - Method in class ai.djl.training.loss.AbstractCompositeLoss
Adds an accumulator for the results of the evaluation with the given key.
addAccumulator(String) - Method in class ai.djl.training.loss.Loss
Adds an accumulator for the results of the evaluation with the given key.
addAll(List<List<String>>) - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Adds the given list of sentences to the DefaultVocabulary.
addAll(NDList) - Method in class ai.djl.ndarray.NDList
Appends all of the NDArrays in the specified NDList to the end of this NDList, in the order that they are returned by the specified NDList's iterator.
addAll(Shape) - Method in class ai.djl.ndarray.types.Shape
Joins this shape with specified other shape.
addAll(Block...) - Method in class ai.djl.nn.ParallelBlock
Adds an array of blocks, each of which is a parallel branch.
addAll(Collection<Block>) - Method in class ai.djl.nn.ParallelBlock
Adds a Collection of blocks, each of which is a parallel branch.
addAll(Block...) - Method in class ai.djl.nn.SequentialBlock
Adds an array of blocks to be executed in sequence, in order.
addAll(Collection<Block>) - Method in class ai.djl.nn.SequentialBlock
Adds a Collection of blocks to be executed in sequence, in order.
addAllDim() - Method in class ai.djl.ndarray.index.NDIndex
Appends a new index to get all values in the dimension.
addAllDim(int) - Method in class ai.djl.ndarray.index.NDIndex
Appends multiple new index to get all values in the dimension.
addArtifact(Artifact) - Method in class ai.djl.repository.Metadata
Adds one artifact for the metadata.
addBooleanIndex(NDArray) - Method in class ai.djl.ndarray.index.NDIndex
Updates the NDIndex by appending a boolean NDArray.
addChildBlock(String, B) - Method in class ai.djl.nn.AbstractBlock
Use this to add a child block to this block.
addEvaluator(Evaluator) - Method in class ai.djl.training.DefaultTrainingConfig
Adds an Evaluator that needs to be computed during training.
addFromCustomizedFile(URL, Function<URL, List<String>>) - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Adds a customized vocabulary to the DefaultVocabulary.
addFromTextFile(Path) - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Adds a text vocabulary to the DefaultVocabulary.
addFromTextFile(URL) - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Adds a text vocabulary to the DefaultVocabulary.
addi(Number) - Method in interface ai.djl.ndarray.NDArray
Adds a number to this NDArray element-wise in place.
addi(NDArray) - Method in interface ai.djl.ndarray.NDArray
Adds other NDArrays to this NDArray element-wise in place.
addi(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Adds a number to this NDArray element-wise in place.
addi(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Adds other NDArrays to this NDArray element-wise in place.
addi(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Adds a number to the NDArray element-wise in place.
addi(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Adds a NDArray to a number element-wise in place.
addi(NDArray...) - Static method in class ai.djl.ndarray.NDArrays
Adds all of the NDArrays together element-wise in place.
addIndices(String, Object...) - Method in class ai.djl.ndarray.index.NDIndex
Updates the NDIndex by appending indices to the array.
addIndices(long...) - Method in class ai.djl.ndarray.index.NDIndex
Updates the NDIndex by appending indices as specified values on the NDArray.
addLicense(License) - Method in class ai.djl.repository.Metadata
Adds one License.
addLoss(Loss) - Method in class ai.djl.training.loss.SimpleCompositeLoss
Adds a Loss that applies to all labels and predictions to this composite loss.
addLoss(Loss, int) - Method in class ai.djl.training.loss.SimpleCompositeLoss
Adds a Loss that applies to a single index of the label and predictions to this composite loss.
addMetric(Metric) - Method in class ai.djl.metric.Metrics
Adds a Metric to the collection.
addMetric(String, Number) - Method in class ai.djl.metric.Metrics
Adds a Metric given the metric's name and value.
addMetric(String, Number, String) - Method in class ai.djl.metric.Metrics
Adds a Metric given the metric's name, value, and unit.
addMetric(String, long) - Method in class ai.djl.training.Trainer
Helper to add a metric for a time difference.
addPad(int, int, NDArraySupplier) - Method in class ai.djl.translate.PaddingStackBatchifier.Builder
Adds a new dimension to be padded in the input NDList.
addPad(int, int, NDArraySupplier, int) - Method in class ai.djl.translate.PaddingStackBatchifier.Builder
Adds a new dimension to be padded in the input NDList.
addParameter(P) - Method in class ai.djl.nn.AbstractBlock
Adds a parameter to this block.
addPickDim(NDArray) - Method in class ai.djl.ndarray.index.NDIndex
Appends a picking index that gets values by index in the axis.
addProperty(String, String) - Method in class ai.djl.modality.Input
Adds a property to the input.
addResource(MRL) - Method in class ai.djl.repository.AbstractRepository
Adds resource to the repository.
addResource(MRL) - Method in class ai.djl.repository.RemoteRepository
Adds resource to the repository.
addResource(MRL) - Method in interface ai.djl.repository.Repository
Adds resource to the repository.
addSingleton(Function<NDArray, NDArray>) - Method in class ai.djl.nn.SequentialBlock
Adds a LambdaBlock.singleton(Function) that applies the given function to the sequence of blocks.
addSliceDim(long, long) - Method in class ai.djl.ndarray.index.NDIndex
Appends a new index to slice the dimension and returns a range of values.
addSliceDim(long, long, long) - Method in class ai.djl.ndarray.index.NDIndex
Appends a new index to slice the dimension and returns a range of values.
addStep(RlEnv.Step) - Method in class ai.djl.modality.rl.LruReplayBuffer
Adds a new step to the buffer.
addStep(RlEnv.Step) - Method in interface ai.djl.modality.rl.ReplayBuffer
Adds a new step to the buffer.
addTargetTransform(Transform) - Method in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
Adds the Transform to the target Pipeline to be applied on the labels.
addTrainingListeners(TrainingListener...) - Method in class ai.djl.training.DefaultTrainingConfig
Adds TrainingListeners for training.
addTransform(Transform) - Method in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
Adds the Transform to the Pipeline use for pre-processing the image.
addTransform(Transform) - Method in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
Adds the Transform to the Pipeline to be applied on the data.
ai.djl - package ai.djl
Contains top level, engine-agnostic classes for both inference and training.
ai.djl.engine - package ai.djl.engine
Contains classes responsible for loading a deep learning engine.
ai.djl.inference - package ai.djl.inference
Contains classes to implement inference tasks.
ai.djl.metric - package ai.djl.metric
Contains classes to collect metrics information.
ai.djl.modality - package ai.djl.modality
Contains utility classes for each of the predefined modalities.
ai.djl.modality.cv - package ai.djl.modality.cv
Contains utility classes for computer vision tasks and image processing.
ai.djl.modality.cv.output - package ai.djl.modality.cv.output
Contains output types used in various computer vision applications.
ai.djl.modality.cv.transform - package ai.djl.modality.cv.transform
Contains Transforms for working with Images.
ai.djl.modality.cv.translator - package ai.djl.modality.cv.translator
Contains translators used for computer vision applications.
ai.djl.modality.cv.translator.wrapper - package ai.djl.modality.cv.translator.wrapper
Contains wrappers to for multiple input formats to a BaseImageTranslator.
ai.djl.modality.cv.util - package ai.djl.modality.cv.util
Contains utility classes for image manipulation.
ai.djl.modality.nlp - package ai.djl.modality.nlp
Contains utility classes for natural language processing tasks.
ai.djl.modality.nlp.bert - package ai.djl.modality.nlp.bert
Contains classes that deal with BERT for natural language pre-processing tasks.
ai.djl.modality.nlp.embedding - package ai.djl.modality.nlp.embedding
Contains classes that deal with word embeddings for natural language pre-processing tasks.
ai.djl.modality.nlp.preprocess - package ai.djl.modality.nlp.preprocess
Contains utility classes for natural language pre-processing tasks.
ai.djl.modality.nlp.qa - package ai.djl.modality.nlp.qa
Contains utility classes for question and answer processing.
ai.djl.modality.nlp.translator - package ai.djl.modality.nlp.translator
Contains utility classes for each of the predefined translator.
ai.djl.modality.rl - package ai.djl.modality.rl
Contains utility classes for reinforcement learning.
ai.djl.modality.rl.agent - package ai.djl.modality.rl.agent
Contains agents to learn using reinforcement learning.
ai.djl.modality.rl.env - package ai.djl.modality.rl.env
Contains environments to train reinforcement learning in.
ai.djl.ndarray - package ai.djl.ndarray
Contains classes and interfaces that define an n-dimensional array.
ai.djl.ndarray.index - package ai.djl.ndarray.index
Contains classes that help access NDArray's indices.
ai.djl.ndarray.index.dim - package ai.djl.ndarray.index.dim
Contains classes that represent an index element in a NDArray's indices.
ai.djl.ndarray.index.full - package ai.djl.ndarray.index.full
Contains classes that represent simplified representations of an NDArray's indices.
ai.djl.ndarray.types - package ai.djl.ndarray.types
Contains classes that define n-dimensional array data types.
ai.djl.nn - package ai.djl.nn
Contains classes to construct neural networks.
ai.djl.nn.convolutional - package ai.djl.nn.convolutional
Contains classes that define convolutional operations extending Convolution and Deconvolution.
ai.djl.nn.core - package ai.djl.nn.core
Contains classes that define simple neural network operations.
ai.djl.nn.norm - package ai.djl.nn.norm
Contains classes that define normalizing neural network operations.
ai.djl.nn.pooling - package ai.djl.nn.pooling
Contains pooling neural network operations in Pool and helpers for it.
ai.djl.nn.recurrent - package ai.djl.nn.recurrent
Contains classes for recurrent neural network operations.
ai.djl.nn.transformer - package ai.djl.nn.transformer
Contains blocks for transformer models.
ai.djl.repository - package ai.djl.repository
Contains a Maven-based Repository format for creating repositories of artifacts such as datasets and model zoos.
ai.djl.repository.zoo - package ai.djl.repository.zoo
Contains classes for working with model zoo repositories.
ai.djl.training - package ai.djl.training
Contains classes and implementations to train a neural network.
ai.djl.training.dataset - package ai.djl.training.dataset
Contains classes to download and prepare training and testing data.
ai.djl.training.evaluator - package ai.djl.training.evaluator
Contains classes for evaluating the effectiveness of models.
ai.djl.training.hyperparameter - package ai.djl.training.hyperparameter
Contains utilities to train, describe, and manipulate Hyperparameters.
ai.djl.training.hyperparameter.optimizer - package ai.djl.training.hyperparameter.optimizer
Contains classes to optimize Hyperparameters.
ai.djl.training.hyperparameter.param - package ai.djl.training.hyperparameter.param
Contains different types of Hyperparameters.
ai.djl.training.initializer - package ai.djl.training.initializer
Contains classes for initializing a Block's Parameters.
ai.djl.training.listener - package ai.djl.training.listener
Contains classes providing functionality during training through TrainingListener.
ai.djl.training.loss - package ai.djl.training.loss
Contains classes for measuring the Loss of a model.
ai.djl.training.optimizer - package ai.djl.training.optimizer
Contains classes for optimizing a neural network Block.
ai.djl.training.tracker - package ai.djl.training.tracker
Contains classes for having a gradually changing hyper-parameter.
ai.djl.training.util - package ai.djl.training.util
Contains utilities to use during training.
ai.djl.translate - package ai.djl.translate
Contains classes and interfaces that translate between java objects and NDArrays.
all() - Method in interface ai.djl.ndarray.NDArray
Returns true if all elements within this NDArray are non-zero or true.
allClose(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns true if two NDArrays are element-wise equal within a tolerance.
allClose(NDArray, double, double, boolean) - Method in interface ai.djl.ndarray.NDArray
Returns true if two NDArray are element-wise equal within a tolerance.
allClose(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns true if two NDArray are element-wise equal within a tolerance.
allClose(NDArray, NDArray, double, double, boolean) - Static method in class ai.djl.ndarray.NDArrays
Returns true if two NDArray are element-wise equal within a tolerance.
allocateDirect(int) - Method in interface ai.djl.ndarray.NDManager
Allocates a new engine specific direct byte buffer.
alternativeManager - Variable in class ai.djl.ndarray.NDArrayAdapter
 
ANY - Static variable in interface ai.djl.Application.CV
Any computer vision application, including those in Application.CV.
ANY - Static variable in interface ai.djl.Application.NLP
Any NLP application, including those in Application.NLP.
ANY - Static variable in interface ai.djl.Application.Tabular
Any tabular application, including those in Application.Tabular.
any() - Method in interface ai.djl.ndarray.NDArray
Returns true if any of the elements within this NDArray are non-zero or true.
apache() - Static method in class ai.djl.repository.License
The default Apache License.
Application - Class in ai.djl
A class contains common tasks that can be completed using deep learning.
Application.CV - Interface in ai.djl
The common set of applications for computer vision (image and video data).
Application.NLP - Interface in ai.djl
The common set of applications for natural language processing (text data).
Application.Tabular - Interface in ai.djl
The common set of applications for tabular data.
arange(float, float, float, DataType) - Method in class ai.djl.ndarray.BaseNDManager
Returns evenly spaced values within a given interval.
arange(int) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced values starting from 0.
arange(float) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced values starting from 0.
arange(int, int) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced values within a given interval with step 1.
arange(float, float) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced values within a given interval with step 1.
arange(int, int, int) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced values within a given interval.
arange(float, float, float) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced values within a given interval.
arange(int, int, int, DataType) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced values within a given interval.
arange(float, float, float, DataType) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced values within a given interval.
arange(float, float, float, DataType, Device) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced values within a given interval.
argMax() - Method in interface ai.djl.ndarray.NDArray
Returns the indices of the maximum values into the flattened NDArray.
argMax(int) - Method in interface ai.djl.ndarray.NDArray
Returns the indices of the maximum values along given axis.
argMax() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the indices of the maximum values into the flattened NDArray.
argMax(int) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the indices of the maximum values along given axis.
argMin() - Method in interface ai.djl.ndarray.NDArray
Returns the indices of the minimum values into the flattened NDArray.
argMin(int) - Method in interface ai.djl.ndarray.NDArray
Returns the indices of the minimum values along given axis.
argMin() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the indices of the minimum values into the flattened NDArray.
argMin(int) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the indices of the minimum values along given axis.
argSort() - Method in interface ai.djl.ndarray.NDArray
Returns the indices that would sort this NDArray.
argSort(int) - Method in interface ai.djl.ndarray.NDArray
Returns the indices that would sort this NDArray given the axis.
argSort(int, boolean) - Method in interface ai.djl.ndarray.NDArray
Returns the indices that would sort this NDArray given the axis.
argSort(int, boolean) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the indices that would sort this NDArray given the axis.
arguments - Variable in class ai.djl.repository.AbstractRepository
 
ArgumentsUtil - Class in ai.djl.translate
A utility class to extract data from model's arguments.
ArrayDataset - Class in ai.djl.training.dataset
ArrayDataset is an implementation of RandomAccessDataset that consist entirely of large NDArrays.
ArrayDataset(RandomAccessDataset.BaseBuilder<?>) - Constructor for class ai.djl.training.dataset.ArrayDataset
Creates a new instance of ArrayDataset with the arguments in ArrayDataset.Builder.
ArrayDataset.Builder - Class in ai.djl.training.dataset
The Builder to construct an ArrayDataset.
Artifact - Class in ai.djl.repository
An Artifact is a set of data files such as a model or dataset.
Artifact() - Constructor for class ai.djl.repository.Artifact
 
Artifact.Item - Class in ai.djl.repository
A file (possibly compressed) within an Artifact.
Artifact.VersionComparator - Class in ai.djl.repository
A Comparator to compare artifacts based on their version numbers.
artifactId - Variable in class ai.djl.repository.Metadata
 
artifacts - Variable in class ai.djl.BaseModel
 
artifacts - Variable in class ai.djl.repository.Metadata
 
asDataType(ByteBuffer) - Method in enum ai.djl.ndarray.types.DataType
Converts a ByteBuffer to a buffer for this data type.
asin() - Method in interface ai.djl.ndarray.NDArray
Returns the inverse trigonometric sine of this NDArray element-wise.
asin() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the inverse trigonometric sine of this NDArray element-wise.
asinh() - Method in interface ai.djl.ndarray.NDArray
Returns the inverse hyperbolic sine of this NDArray element-wise.
asinh() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the inverse hyperbolic sine of this NDArray element-wise.
asNumpy() - Method in enum ai.djl.ndarray.types.DataType
Returns a numpy string value.
atan() - Method in interface ai.djl.ndarray.NDArray
Returns the inverse trigonometric tangent of this NDArray element-wise.
atan() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the inverse trigonometric tangent of this NDArray element-wise.
atanh() - Method in interface ai.djl.ndarray.NDArray
Returns the inverse hyperbolic tangent of this NDArray element-wise.
atanh() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the inverse hyperbolic tangent of this NDArray element-wise.
attach(NDManager) - Method in class ai.djl.ndarray.NDArrayAdapter
Attaches this NDResource to the specified NDManager.
attach(NDManager) - Method in class ai.djl.ndarray.NDList
Attaches this NDResource to the specified NDManager.
attach(NDManager) - Method in interface ai.djl.ndarray.NDResource
Attaches this NDResource to the specified NDManager.
attachAll(NDResource...) - Method in interface ai.djl.ndarray.NDManager
Attaches all resources to this manager.
attachInternal(String, AutoCloseable) - Method in class ai.djl.ndarray.BaseNDManager
Attaches a resource to this NDManager.
attachInternal(String, AutoCloseable) - Method in interface ai.djl.ndarray.NDManager
Attaches a resource to this NDManager.
availableSize() - Method in class ai.djl.training.dataset.ArrayDataset
Returns the number of records available to be read in this Dataset.
availableSize() - Method in class ai.djl.training.dataset.RandomAccessDataset
Returns the number of records available to be read in this Dataset.
avgPool1d(NDArray, Shape, Shape, Shape, boolean, boolean) - Static method in class ai.djl.nn.pooling.Pool
Performs 1-D Avg Pooling on the input.
avgPool1dBlock(Shape, Shape, Shape, boolean, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool1dBlock pooling function in its forward function.
avgPool1dBlock(Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool1dBlock pooling function in its forward function.
avgPool1dBlock(Shape, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool1dBlock pooling function in its forward function.
avgPool1dBlock(Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool1dBlock pooling function in its forward function.
avgPool1dBlock(Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool1dBlock pooling function in its forward function.
avgPool2d(NDArray, Shape, Shape, Shape, boolean, boolean) - Static method in class ai.djl.nn.pooling.Pool
Performs 2-D Avg Pooling on the input.
avgPool2dBlock(Shape, Shape, Shape, boolean, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool2dBlock pooling function in its forward function.
avgPool2dBlock(Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool2dBlock pooling function in its forward function.
avgPool2dBlock(Shape, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool2dBlock pooling function in its forward function.
avgPool2dBlock(Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool2dBlock pooling function in its forward function.
avgPool2dBlock(Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool2dBlock pooling function in its forward function.
avgPool3d(NDArray, Shape, Shape, Shape, boolean, boolean) - Static method in class ai.djl.nn.pooling.Pool
Performs 3-D Avg Pooling on the input.
avgPool3dBlock(Shape, Shape, Shape, boolean, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool3dBlock pooling function in its forward function.
avgPool3dBlock(Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool3dBlock pooling function in its forward function.
avgPool3dBlock(Shape, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool3dBlock pooling function in its forward function.
avgPool3dBlock(Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool3dBlock pooling function in its forward function.
avgPool3dBlock(Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the avgPool3dBlock pooling function in its forward function.
axis(int...) - Method in class ai.djl.nn.norm.LayerNorm.Builder
List the axis over which the mean and variance will be calculated (alternative to normalizedShape).
axis - Variable in class ai.djl.training.evaluator.AbstractAccuracy
 

B

backward(NDArray) - Method in interface ai.djl.training.GradientCollector
Calculate the gradient w.r.t previously marked variable (head).
base() - Method in class ai.djl.nn.transformer.BertBlock.Builder
Sets this builder's params to the BASE config of the original BERT paper.
BaseBuilder() - Constructor for class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
 
BaseBuilder() - Constructor for class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
 
BaseBuilder() - Constructor for class ai.djl.nn.core.Embedding.BaseBuilder
 
BaseBuilder() - Constructor for class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
 
BaseBuilder() - Constructor for class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
 
BaseHpOptimizer - Class in ai.djl.training.hyperparameter.optimizer
A base containing shared implementations for HpOptimizers.
BaseHpOptimizer(HpSet) - Constructor for class ai.djl.training.hyperparameter.optimizer.BaseHpOptimizer
Constructs a BaseHpOptimizer.
BaseImageTranslator<T> - Class in ai.djl.modality.cv.translator
Built-in Translator that provides default image pre-processing.
BaseImageTranslator(BaseImageTranslator.BaseBuilder<?>) - Constructor for class ai.djl.modality.cv.translator.BaseImageTranslator
Constructs an ImageTranslator with the provided builder.
BaseImageTranslator.BaseBuilder<T extends BaseImageTranslator.BaseBuilder> - Class in ai.djl.modality.cv.translator
A builder to extend for all classes extending the BaseImageTranslator.
BaseImageTranslator.ClassificationBuilder<T extends BaseImageTranslator.BaseBuilder> - Class in ai.djl.modality.cv.translator
A Builder to construct a ImageClassificationTranslator.
BaseImageTranslator.SynsetLoader - Class in ai.djl.modality.cv.translator
 
BaseModel - Class in ai.djl
BaseModel is the basic implementation of Model.
BaseModel(String) - Constructor for class ai.djl.BaseModel
 
BaseModelLoader - Class in ai.djl.repository.zoo
Shared code for the ModelLoader implementations.
BaseModelLoader(MRL) - Constructor for class ai.djl.repository.zoo.BaseModelLoader
Constructs a ModelLoader given the repository, mrl, and version.
BaseNDManager - Class in ai.djl.ndarray
BaseNDManager is the default implementation of NDManager.
BaseNDManager(NDManager, Device) - Constructor for class ai.djl.ndarray.BaseNDManager
 
BaseNDManager.TempResource - Class in ai.djl.ndarray
 
basic() - Static method in interface ai.djl.training.listener.TrainingListener.Defaults
A basic TrainingListener set with minimal recommended functionality.
Batch - Class in ai.djl.training.dataset
A Batch is used to hold multiple items (data and label pairs) from a Dataset.
Batch(NDManager, NDList, NDList, int, Batchifier, Batchifier, long, long) - Constructor for class ai.djl.training.dataset.Batch
Creates a new instance of Batch with the given manager, data and labels.
BatchData(Batch, Map<Device, NDList>, Map<Device, NDList>) - Constructor for class ai.djl.training.listener.TrainingListener.BatchData
Constructs a new TrainingListener.BatchData.
batchDot(NDArray) - Method in interface ai.djl.ndarray.NDArray
Batchwise product of this NDArray and the other NDArray.
batchDot(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Batchwise product of this NDArray and the other NDArray.
batchFirst - Variable in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
 
batchFirst - Variable in class ai.djl.nn.recurrent.RecurrentBlock
 
batchFlatten(NDArray) - Static method in class ai.djl.nn.Blocks
Inflates the NDArray provided as input to a 2-D NDArray of shape (batch, size).
batchFlatten(NDArray, long) - Static method in class ai.djl.nn.Blocks
Inflates the NDArray provided as input to a 2-D NDArray of shape (batch, size).
batchFlattenBlock() - Static method in class ai.djl.nn.Blocks
Creates a Block whose forward function applies the batchFlatten method.
batchFlattenBlock(long) - Static method in class ai.djl.nn.Blocks
Creates a Block whose forward function applies the batchFlatten method.
batchifier - Variable in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
 
batchifier - Variable in class ai.djl.modality.nlp.translator.QATranslator
 
Batchifier - Interface in ai.djl.translate
An interface that provides methods to convert an un-batched NDList into a batched NDList and vice versa.
batchify(NDList[]) - Method in interface ai.djl.translate.Batchifier
Converts an array of NDList into an NDList.
batchify(NDList[]) - Method in class ai.djl.translate.PaddingStackBatchifier
Converts an array of NDList into an NDList.
batchify(NDList[]) - Method in class ai.djl.translate.StackBatchifier
Converts an array of NDList into an NDList.
BatchNorm - Class in ai.djl.nn.norm
In batch training (training with more than one samples per iteration), a batch normalization layer works by normalizing the values of input data to have mean of 0 and variance of 1.
batchNorm(NDArray, NDArray, NDArray) - Static method in class ai.djl.nn.norm.BatchNorm
Applies Batch Normalization for each channel across a batch of data.
batchNorm(NDArray, NDArray, NDArray, NDArray, NDArray) - Static method in class ai.djl.nn.norm.BatchNorm
Applies Batch Normalization for each channel across a batch of data.
batchNorm(NDArray, NDArray, NDArray, NDArray, NDArray, int) - Static method in class ai.djl.nn.norm.BatchNorm
Applies Batch Normalization for each channel across a batch of data.
batchNorm(NDArray, NDArray, NDArray, NDArray, NDArray, int, float, float, boolean) - Static method in class ai.djl.nn.norm.BatchNorm
Applies Batch Normalization for each channel across a batch of data.
BatchNorm.Builder - Class in ai.djl.nn.norm
The Builder to construct a BatchNorm.
batchPredict(List<I>) - Method in class ai.djl.inference.Predictor
Predicts a batch for inference.
BatchSampler - Class in ai.djl.training.dataset
BatchSampler is a Sampler that returns a single epoch over the data.
BatchSampler(Sampler.SubSampler, int) - Constructor for class ai.djl.training.dataset.BatchSampler
Creates a new instance of BatchSampler that samples from the given Sampler.SubSampler, and yields a mini-batch of indices.
BatchSampler(Sampler.SubSampler, int, boolean) - Constructor for class ai.djl.training.dataset.BatchSampler
Creates a new instance of BatchSampler that samples from the given Sampler.SubSampler, and yields a mini-batch of indices.
beforeInitialize(Shape...) - Method in class ai.djl.nn.AbstractBlock
Performs any action necessary before initialization.
beforeInitialize(Shape...) - Method in class ai.djl.nn.convolutional.Convolution
Performs any action necessary before initialization.
beforeInitialize(Shape...) - Method in class ai.djl.nn.convolutional.Deconvolution
Performs any action necessary before initialization.
beforeInitialize(Shape...) - Method in class ai.djl.nn.core.Linear
Performs any action necessary before initialization.
beforeInitialize(Shape...) - Method in class ai.djl.nn.norm.BatchNorm
Performs any action necessary before initialization.
beforeInitialize(Shape...) - Method in class ai.djl.nn.norm.LayerNorm
Performs any action necessary before initialization.
beforeInitialize(Shape...) - Method in class ai.djl.nn.recurrent.RecurrentBlock
Performs any action necessary before initialization.
BertBlock - Class in ai.djl.nn.transformer
Implements the core bert model (without next sentence and masked language task) of bert.
BertBlock.Builder - Class in ai.djl.nn.transformer
The Builder to construct a BertBlock type of Block.
BertFullTokenizer - Class in ai.djl.modality.nlp.bert
BertFullTokenizer runs end to end tokenization of input text
BertFullTokenizer(Vocabulary, boolean) - Constructor for class ai.djl.modality.nlp.bert.BertFullTokenizer
Creates an instance of BertFullTokenizer.
BertMaskedLanguageModelBlock - Class in ai.djl.nn.transformer
Block for the bert masked language task.
BertMaskedLanguageModelBlock(BertBlock, Function<NDArray, NDArray>) - Constructor for class ai.djl.nn.transformer.BertMaskedLanguageModelBlock
Creates a new block that applies the masked language task.
BertMaskedLanguageModelLoss - Class in ai.djl.nn.transformer
The loss for the bert masked language model task.
BertMaskedLanguageModelLoss(int, int, int) - Constructor for class ai.djl.nn.transformer.BertMaskedLanguageModelLoss
Creates an MLM loss.
BertNextSentenceBlock - Class in ai.djl.nn.transformer
Block to perform the Bert next-sentence-prediction task.
BertNextSentenceBlock() - Constructor for class ai.djl.nn.transformer.BertNextSentenceBlock
Creates a next sentence block.
BertNextSentenceLoss - Class in ai.djl.nn.transformer
Calculates the loss for the next sentence prediction task.
BertNextSentenceLoss(int, int) - Constructor for class ai.djl.nn.transformer.BertNextSentenceLoss
Creates a new bert next sentence loss.
BertPretrainingBlock - Class in ai.djl.nn.transformer
Creates a block that performs all bert pretraining tasks (next sentence and masked language).
BertPretrainingBlock(BertBlock.Builder) - Constructor for class ai.djl.nn.transformer.BertPretrainingBlock
Creates a new Bert pretraining block fitting the given bert configuration.
BertPretrainingLoss - Class in ai.djl.nn.transformer
Loss that combines the next sentence and masked language losses of bert pretraining.
BertPretrainingLoss() - Constructor for class ai.djl.nn.transformer.BertPretrainingLoss
Creates a loss combining the next sentence and masked language loss for bert pretraining.
BertToken - Class in ai.djl.modality.nlp.bert
BertToken contains all the information for Bert model after encoding question and paragraph.
BertToken(List<String>, List<Long>, List<Long>, int) - Constructor for class ai.djl.modality.nlp.bert.BertToken
Creates an instance of BertToken which includes information for Bert model.
BertTokenizer - Class in ai.djl.modality.nlp.bert
BertTokenizer is a class to help you encode question and paragraph sentence.
BertTokenizer() - Constructor for class ai.djl.modality.nlp.bert.BertTokenizer
 
best() - Method in class ai.djl.modality.Classifications
Returns the most likely class for the classification.
bias - Variable in class ai.djl.nn.convolutional.Convolution
 
bias - Variable in class ai.djl.nn.convolutional.Deconvolution
 
bidirectional - Variable in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
 
bidirectional - Variable in class ai.djl.nn.recurrent.RecurrentBlock
 
BigGANTranslator - Class in ai.djl.modality.cv.translator
Built-in Translator that provides preprocessing and postprocessing for BigGAN.
BigGANTranslator(float) - Constructor for class ai.djl.modality.cv.translator.BigGANTranslator
Constructs a translator for BigGAN.
BigGANTranslatorFactory - Class in ai.djl.modality.cv.translator
A TranslatorFactory that creates a BigGANTranslator instance.
BigGANTranslatorFactory() - Constructor for class ai.djl.modality.cv.translator.BigGANTranslatorFactory
 
BinaryAccuracy - Class in ai.djl.training.evaluator
BinaryAccuracy is the AbstractAccuracy with two classes.
BinaryAccuracy(String, float, int, int) - Constructor for class ai.djl.training.evaluator.BinaryAccuracy
Creates a binary (two class) accuracy evaluator.
BinaryAccuracy(String, float, int) - Constructor for class ai.djl.training.evaluator.BinaryAccuracy
Creates a binary (two class) accuracy evaluator that computes accuracy across axis 1 along given index.
BinaryAccuracy(float) - Constructor for class ai.djl.training.evaluator.BinaryAccuracy
Creates a binary (two class) accuracy evaluator that computes accuracy across axis 1 along the 0th index.
BinaryAccuracy() - Constructor for class ai.djl.training.evaluator.BinaryAccuracy
Creates a binary (two class) accuracy evaluator with 0 threshold.
block - Variable in class ai.djl.BaseModel
 
block - Variable in class ai.djl.inference.Predictor
 
block - Variable in class ai.djl.modality.nlp.Decoder
 
block - Variable in class ai.djl.modality.nlp.Encoder
 
Block - Interface in ai.djl.nn
A Block is a composable function that forms a neural network.
BlockFactory - Interface in ai.djl.nn
Block factory is a component to make standard for block creating and saving procedure.
BlockList - Class in ai.djl.nn
Represents a set of names and Blocks.
BlockList() - Constructor for class ai.djl.nn.BlockList
Creates an empty BlockList.
BlockList(int) - Constructor for class ai.djl.nn.BlockList
Constructs an empty BlockList with the specified initial capacity.
BlockList(List<String>, List<Block>) - Constructor for class ai.djl.nn.BlockList
Constructs a BlockList containing the elements of the specified keys and values.
BlockList(List<Pair<String, Block>>) - Constructor for class ai.djl.nn.BlockList
Constructs a BlockList containing the elements of the specified list of Pairs.
BlockList(Map<String, Block>) - Constructor for class ai.djl.nn.BlockList
Constructs a BlockList containing the elements of the specified map.
Blocks - Class in ai.djl.nn
Utility class that provides some useful blocks.
booleanMask(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns portion of this NDArray given the index boolean NDArray along first axis.
booleanMask(NDArray, int) - Method in interface ai.djl.ndarray.NDArray
Returns portion of this NDArray given the index boolean NDArray along given axis.
booleanMask(NDArray, int) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns portion of this NDArray given the index boolean NDArray along given axis.
booleanMask(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns portion of the NDArray given the index boolean NDArray along first axis.
booleanMask(NDArray, NDArray, int) - Static method in class ai.djl.ndarray.NDArrays
Returns portion of the NDArray given the index boolean NDArray along given axis.
booleanValue(Map<String, ?>, String) - Static method in class ai.djl.translate.ArgumentsUtil
Returns the boolean value from the arguments.
booleanValue(Map<String, ?>, String, boolean) - Static method in class ai.djl.translate.ArgumentsUtil
Returns the boolean value from the arguments.
BoundingBox - Interface in ai.djl.modality.cv.output
An interface representing a bounding box around an object inside an image.
BoundingBoxError - Class in ai.djl.training.evaluator
BoundingBoxError is an Evaluator that computes the error in the prediction of bounding boxes in SingleShotDetection model.
BoundingBoxError(String) - Constructor for class ai.djl.training.evaluator.BoundingBoxError
Creates an BoundingBoxError evaluator.
boxIntersection(Rectangle, Rectangle) - Method in class ai.djl.modality.cv.translator.YoloV5Translator
 
boxIou(Rectangle, Rectangle) - Method in class ai.djl.modality.cv.translator.YoloV5Translator
 
boxUnion(Rectangle, Rectangle) - Method in class ai.djl.modality.cv.translator.YoloV5Translator
 
broadcast(Shape) - Method in interface ai.djl.ndarray.NDArray
Broadcasts this NDArray to be the given shape.
broadcast(long...) - Method in interface ai.djl.ndarray.NDArray
Broadcasts this NDArray to be the given shape.
broadcast(Shape) - Method in class ai.djl.ndarray.NDArrayAdapter
Broadcasts this NDArray to be the given shape.
BufferedImageFactory - Class in ai.djl.modality.cv
BufferedImageFactory is the default implementation of ImageFactory.
BufferedImageFactory() - Constructor for class ai.djl.modality.cv.BufferedImageFactory
 
build() - Method in class ai.djl.modality.cv.MultiBoxDetection.Builder
Builds a MultiBoxDetection block.
build() - Method in class ai.djl.modality.cv.MultiBoxPrior.Builder
Builds a MultiBoxPrior block.
build() - Method in class ai.djl.modality.cv.MultiBoxTarget.Builder
Builds a MultiBoxTarget block.
build() - Method in class ai.djl.modality.cv.translator.ImageClassificationTranslator.Builder
Builds the ImageClassificationTranslator with the provided data.
build() - Method in class ai.djl.modality.cv.translator.InstanceSegmentationTranslator.Builder
Builds the translator.
build() - Method in class ai.djl.modality.cv.translator.SimplePoseTranslator.Builder
Builds the translator.
build() - Method in class ai.djl.modality.cv.translator.SingleShotDetectionTranslator.Builder
Builds the translator.
build() - Method in class ai.djl.modality.cv.translator.YoloTranslator.Builder
Builds the translator.
build() - Method in class ai.djl.modality.cv.translator.YoloV5Translator.Builder
Builds the translator.
build() - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Builds the DefaultVocabulary object with the set arguments.
build() - Method in class ai.djl.modality.nlp.embedding.TrainableWordEmbedding.Builder
Builds a new instance of TrainableWordEmbedding based on the arguments in this builder.
build() - Method in class ai.djl.nn.convolutional.Conv1d.Builder
Builds a Conv1d block.
build() - Method in class ai.djl.nn.convolutional.Conv1dTranspose.Builder
Builds a Conv1dTranspose block.
build() - Method in class ai.djl.nn.convolutional.Conv2d.Builder
Builds a Conv2d block.
build() - Method in class ai.djl.nn.convolutional.Conv2dTranspose.Builder
Builds a Conv2dTranspose block.
build() - Method in class ai.djl.nn.convolutional.Conv3d.Builder
Builds a Conv3d block.
build() - Method in class ai.djl.nn.core.Linear.Builder
Returns the constructed Linear.
build() - Method in class ai.djl.nn.norm.BatchNorm.Builder
Builds a BatchNorm block.
build() - Method in class ai.djl.nn.norm.Dropout.Builder
Builds a Dropout block.
build() - Method in class ai.djl.nn.norm.LayerNorm.Builder
Builds a LayerNorm block.
build() - Method in class ai.djl.nn.Parameter.Builder
Builds a Parameter instance.
build() - Method in class ai.djl.nn.recurrent.GRU.Builder
Builds a GRU block.
build() - Method in class ai.djl.nn.recurrent.LSTM.Builder
Builds a LSTM block.
build() - Method in class ai.djl.nn.recurrent.RNN.Builder
Builds a RNN block.
build() - Method in class ai.djl.nn.transformer.BertBlock.Builder
Returns a new BertBlock with the parameters of this builder.
build() - Method in class ai.djl.nn.transformer.IdEmbedding.Builder
Builds the IdEmbedding.
build() - Method in class ai.djl.nn.transformer.ScaledDotProductAttentionBlock.Builder
Creates a new ScaledDotProductAttentionBlock with the current configuration.
build() - Method in class ai.djl.repository.zoo.Criteria.Builder
Builds a Criteria instance.
build() - Method in class ai.djl.training.dataset.ArrayDataset.Builder
Builds a new instance of ArrayDataset with the specified data and labels.
build() - Method in class ai.djl.training.optimizer.Adadelta.Builder
Builds a Adadelta block.
build() - Method in class ai.djl.training.optimizer.Adagrad.Builder
Builds a Adagrad block.
build() - Method in class ai.djl.training.optimizer.Adam.Builder
Builds a Adam block.
build() - Method in class ai.djl.training.optimizer.Nag.Builder
Builds a Nag block.
build() - Method in class ai.djl.training.optimizer.RmsProp.Builder
Builds a RmsProp block.
build() - Method in class ai.djl.training.optimizer.Sgd.Builder
Builds a Sgd block.
build() - Method in class ai.djl.training.tracker.CosineTracker.Builder
Builds a CosineTracker block.
build() - Method in class ai.djl.training.tracker.FactorTracker.Builder
Builds a FactorTracker block.
build() - Method in class ai.djl.training.tracker.LinearTracker.Builder
Builds a LinearTracker block.
build() - Method in class ai.djl.training.tracker.MultiFactorTracker.Builder
Builds a MultiFactorTracker block.
build() - Method in class ai.djl.training.tracker.PolynomialDecayTracker.Builder
Builds a PolynomialDecayTracker.
build() - Method in class ai.djl.training.tracker.WarmUpTracker.Builder
Builds a WarmUpTracker block.
build() - Method in class ai.djl.translate.PaddingStackBatchifier.Builder
builder() - Static method in class ai.djl.modality.cv.MultiBoxDetection
Creates a builder to build a MultiBoxDetection.
builder() - Static method in class ai.djl.modality.cv.MultiBoxPrior
Creates a builder to build a MultiBoxPrior.
builder() - Static method in class ai.djl.modality.cv.MultiBoxTarget
Creates a builder to build a MultiBoxTarget.
builder() - Static method in class ai.djl.modality.cv.translator.ImageClassificationTranslator
Creates a builder to build a ImageClassificationTranslator.
builder(Map<String, ?>) - Static method in class ai.djl.modality.cv.translator.ImageClassificationTranslator
Creates a builder to build a ImageClassificationTranslator with specified arguments.
builder() - Static method in class ai.djl.modality.cv.translator.InstanceSegmentationTranslator
Creates a builder to build a InstanceSegmentationTranslator.
builder(Map<String, ?>) - Static method in class ai.djl.modality.cv.translator.InstanceSegmentationTranslator
Creates a builder to build a InstanceSegmentationTranslator with specified arguments.
builder() - Static method in class ai.djl.modality.cv.translator.SimplePoseTranslator
Creates a builder to build a SimplePoseTranslator.
builder(Map<String, ?>) - Static method in class ai.djl.modality.cv.translator.SimplePoseTranslator
Creates a builder to build a SimplePoseTranslator with specified arguments.
builder() - Static method in class ai.djl.modality.cv.translator.SingleShotDetectionTranslator
Creates a builder to build a SingleShotDetectionTranslator.
builder(Map<String, ?>) - Static method in class ai.djl.modality.cv.translator.SingleShotDetectionTranslator
Creates a builder to build a SingleShotDetectionTranslator with specified arguments.
Builder() - Constructor for class ai.djl.modality.cv.translator.SingleShotDetectionTranslator.Builder
 
builder() - Static method in class ai.djl.modality.cv.translator.YoloTranslator
Creates a builder to build a YoloTranslator.
builder(Map<String, ?>) - Static method in class ai.djl.modality.cv.translator.YoloTranslator
Creates a builder to build a YoloTranslator with specified arguments.
Builder() - Constructor for class ai.djl.modality.cv.translator.YoloTranslator.Builder
 
builder() - Static method in class ai.djl.modality.cv.translator.YoloV5Translator
Creates a builder to build a YoloV5Translator.
builder(Map<String, ?>) - Static method in class ai.djl.modality.cv.translator.YoloV5Translator
Creates a builder to build a YoloV5Translator with specified arguments.
Builder() - Constructor for class ai.djl.modality.cv.translator.YoloV5Translator.Builder
 
builder() - Static method in class ai.djl.modality.nlp.DefaultVocabulary
Creates a new builder to build a DefaultVocabulary.
builder() - Static method in class ai.djl.modality.nlp.embedding.TrainableWordEmbedding
Creates a builder to build an Embedding.
builder() - Static method in class ai.djl.nn.convolutional.Conv1d
Creates a builder to build a Conv1d.
builder() - Static method in class ai.djl.nn.convolutional.Conv1dTranspose
Creates a builder to build a Conv1dTranspose.
builder() - Static method in class ai.djl.nn.convolutional.Conv2d
Creates a builder to build a Conv2d.
builder() - Static method in class ai.djl.nn.convolutional.Conv2dTranspose
Creates a builder to build a Conv2dTranspose.
builder() - Static method in class ai.djl.nn.convolutional.Conv3d
Creates a builder to build a Conv3d.
builder() - Static method in class ai.djl.nn.core.Linear
Creates a builder to build a Linear.
builder() - Static method in class ai.djl.nn.norm.BatchNorm
Creates a builder to build a BatchNorm.
builder() - Static method in class ai.djl.nn.norm.Dropout
Creates a builder to build a Dropout.
builder() - Static method in class ai.djl.nn.norm.LayerNorm
Creates a builder to build a LayerNorm.
builder() - Static method in class ai.djl.nn.Parameter
Creates a builder to build a Parameter.
Builder() - Constructor for class ai.djl.nn.Parameter.Builder
 
builder() - Static method in class ai.djl.nn.recurrent.GRU
Creates a builder to build a GRU.
Builder() - Constructor for class ai.djl.nn.recurrent.GRU.Builder
 
builder() - Static method in class ai.djl.nn.recurrent.LSTM
Creates a builder to build a LSTM.
Builder() - Constructor for class ai.djl.nn.recurrent.LSTM.Builder
 
builder() - Static method in class ai.djl.nn.recurrent.RNN
Creates a builder to build a RNN.
Builder() - Constructor for class ai.djl.nn.recurrent.RNN.Builder
 
builder() - Static method in class ai.djl.nn.transformer.BertBlock
Returns a new BertBlock builder.
Builder() - Constructor for class ai.djl.nn.transformer.IdEmbedding.Builder
 
builder() - Static method in class ai.djl.nn.transformer.ScaledDotProductAttentionBlock
Creates a new Builder to build an Attention Block with.
builder() - Static method in class ai.djl.repository.zoo.Criteria
Creates a builder to build a Criteria.
Builder() - Constructor for class ai.djl.training.dataset.ArrayDataset.Builder
 
builder() - Static method in class ai.djl.training.optimizer.Adagrad
Creates a builder to build a Adam.
builder() - Static method in class ai.djl.training.optimizer.Adam
Creates a builder to build a Adam.
builder() - Static method in class ai.djl.training.optimizer.RmsProp
Creates a builder to build a RMSProp.
builder() - Static method in class ai.djl.training.tracker.CosineTracker
Creates a new builder.
builder() - Static method in class ai.djl.training.tracker.FactorTracker
Creates a new builder.
builder() - Static method in class ai.djl.training.tracker.LinearTracker
Creates a new builder.
builder() - Static method in class ai.djl.training.tracker.MultiFactorTracker
Creates a new builder.
builder() - Static method in class ai.djl.training.tracker.PolynomialDecayTracker
Creates a new builder.
builder() - Static method in class ai.djl.training.tracker.WarmUpTracker
Creates a new builder.
builder() - Static method in class ai.djl.translate.PaddingStackBatchifier
buildModel(HpSet) - Method in class ai.djl.training.hyperparameter.EasyHpo
Builds the Model and Block to train.
buildSentence(List<String>) - Method in class ai.djl.modality.nlp.preprocess.SimpleTokenizer
Combines a list of tokens to form a sentence.
buildSentence(List<String>) - Method in interface ai.djl.modality.nlp.preprocess.Tokenizer
Combines a list of tokens to form a sentence.
BytesSupplier - Interface in ai.djl.ndarray
Represents a supplier of byte[].

C

cast(DataType) - Method in interface ai.djl.Model
Casts the model to support a different precision level.
cast(DataType) - Method in class ai.djl.nn.AbstractBlock
Guaranteed to throw an exception.
cast(DataType) - Method in interface ai.djl.nn.Block
Guaranteed to throw an exception.
cast(DataType) - Method in class ai.djl.repository.zoo.ZooModel
Casts the model to support a different precision level.
cbrt() - Method in interface ai.djl.ndarray.NDArray
Returns the cube-root of this NDArray element-wise.
cbrt() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the cube-root of this NDArray element-wise.
ceil() - Method in interface ai.djl.ndarray.NDArray
Returns the ceiling of this NDArray element-wise.
ceil() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the ceiling of this NDArray element-wise.
CenterCrop - Class in ai.djl.modality.cv.transform
A Transform that crops the center of an image.
CenterCrop() - Constructor for class ai.djl.modality.cv.transform.CenterCrop
Creates a CenterCrop Transform that crops to size min(width, height).
CenterCrop(int, int) - Constructor for class ai.djl.modality.cv.transform.CenterCrop
Creates a CenterCrop Transform that crops the given width and height.
centerCrop(NDArray) - Static method in class ai.djl.modality.cv.util.NDImageUtils
Crops an image to a square of size min(width, height).
centerCrop(NDArray, int, int) - Static method in class ai.djl.modality.cv.util.NDImageUtils
Crops an image to a given width and height from the center of the image.
checkConcatInput(NDList) - Static method in class ai.djl.ndarray.NDUtils
Check two criteria of concat input: 1.
checkLabelShapes(NDArray, NDArray, boolean) - Method in class ai.djl.training.evaluator.Evaluator
Checks if the two input NDArray have the same length or shape.
checkLabelShapes(NDArray, NDArray) - Method in class ai.djl.training.evaluator.Evaluator
Checks the length of NDArrays.
children - Variable in class ai.djl.nn.AbstractBlock
All direct children of this Block.
chooseAction(RlEnv, boolean) - Method in class ai.djl.modality.rl.agent.EpsilonGreedy
Chooses the next action to take within the RlEnv.
chooseAction(RlEnv, boolean) - Method in class ai.djl.modality.rl.agent.QAgent
Chooses the next action to take within the RlEnv.
chooseAction(RlEnv, boolean) - Method in interface ai.djl.modality.rl.agent.RlAgent
Chooses the next action to take within the RlEnv.
classes - Variable in class ai.djl.modality.cv.translator.ObjectDetectionTranslator
 
Classification(String, double) - Constructor for class ai.djl.modality.Classifications.Classification
Constructs a single class result for a classification.
ClassificationBuilder() - Constructor for class ai.djl.modality.cv.translator.BaseImageTranslator.ClassificationBuilder
 
Classifications - Class in ai.djl.modality
Classifications is the container that stores the classification results for classification on a single input.
Classifications(List<String>, List<Double>) - Constructor for class ai.djl.modality.Classifications
Constructs a Classifications using a parallel list of classNames and probabilities.
Classifications(List<String>, NDArray) - Constructor for class ai.djl.modality.Classifications
Constructs a Classifications using list of classNames parallel to an NDArray of probabilities.
Classifications.Classification - Class in ai.djl.modality
A Classification stores the classification result for a single class on a single input.
Classifications.ClassificationsSerializer - Class in ai.djl.modality
A customized Gson serializer to serialize the Classifications object.
ClassificationsSerializer() - Constructor for class ai.djl.modality.Classifications.ClassificationsSerializer
 
classNames - Variable in class ai.djl.modality.Classifications
 
clear() - Method in class ai.djl.nn.AbstractBlock
Closes all the parameters of the block.
clear() - Method in interface ai.djl.nn.Block
Closes all the parameters of the block.
clip(Number, Number) - Method in interface ai.djl.ndarray.NDArray
Clips (limit) the values in this NDArray.
clip(Number, Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Clips (limit) the values in this NDArray.
clipGrad - Variable in class ai.djl.training.optimizer.Optimizer
 
close() - Method in class ai.djl.BaseModel
close() - Method in class ai.djl.inference.Predictor
close() - Method in class ai.djl.modality.nlp.embedding.ModelZooTextEmbedding
close() - Method in interface ai.djl.modality.rl.env.RlEnv
close() - Method in interface ai.djl.modality.rl.env.RlEnv.Step
close() - Method in interface ai.djl.Model
close() - Method in class ai.djl.ndarray.BaseNDManager
close() - Method in interface ai.djl.ndarray.NDArray
close() - Method in class ai.djl.ndarray.NDArrayAdapter
close() - Method in class ai.djl.ndarray.NDList
close() - Method in interface ai.djl.ndarray.NDManager
close() - Method in interface ai.djl.ndarray.NDResource
close() - Method in class ai.djl.nn.Parameter
close() - Method in class ai.djl.repository.zoo.ZooModel
close() - Method in class ai.djl.training.dataset.Batch
close() - Method in interface ai.djl.training.GradientCollector
close() - Method in class ai.djl.training.LocalParameterServer
close() - Method in interface ai.djl.training.ParameterServer
close() - Method in class ai.djl.training.Trainer
close() - Method in class ai.djl.training.util.MinMaxScaler
close() - Method in interface ai.djl.translate.TranslatorContext
closed - Variable in class ai.djl.ndarray.BaseNDManager
 
collectMemoryInfo(Metrics) - Static method in class ai.djl.training.listener.MemoryTrainingListener
Collects memory information.
compare(Artifact, Artifact) - Method in class ai.djl.repository.Artifact.VersionComparator
compareTo(Version) - Method in class ai.djl.repository.Version
components - Variable in class ai.djl.training.loss.AbstractCompositeLoss
 
concat(NDArray) - Method in interface ai.djl.ndarray.NDArray
Joins a NDArray along the first axis.
concat(NDArray, int) - Method in interface ai.djl.ndarray.NDArray
Joins a NDArray along an existing axis.
concat(NDList) - Static method in class ai.djl.ndarray.NDArrays
Joins a NDList along the first axis.
concat(NDList, int) - Static method in class ai.djl.ndarray.NDArrays
Joins a NDList along an existing axis.
configPostProcess(Map<String, ?>) - Method in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
 
configPostProcess(Map<String, ?>) - Method in class ai.djl.modality.cv.translator.BaseImageTranslator.ClassificationBuilder
configPostProcess(Map<String, ?>) - Method in class ai.djl.modality.cv.translator.ImageClassificationTranslator.Builder
configPostProcess(Map<String, ?>) - Method in class ai.djl.modality.cv.translator.InstanceSegmentationTranslator.Builder
configPostProcess(Map<String, ?>) - Method in class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder
configPostProcess(Map<String, ?>) - Method in class ai.djl.modality.cv.translator.SimplePoseTranslator.Builder
configPostProcess(Map<String, ?>) - Method in class ai.djl.modality.cv.translator.YoloV5Translator.Builder
configPreProcess(Map<String, ?>) - Method in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
 
configure(Map<String, ?>) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Configures the builder with the model arguments.
ConstantEmbedding - Class in ai.djl.nn.core
An AbstractIndexedEmbedding that always returns a constant value.
ConstantEmbedding(NDArray) - Constructor for class ai.djl.nn.core.ConstantEmbedding
Constructs a constant embedding with the given constant.
ConstantInitializer - Class in ai.djl.training.initializer
Initializer that generates tensors with constant values.
ConstantInitializer(float) - Constructor for class ai.djl.training.initializer.ConstantInitializer
Creates a Constant Initializer.
contains(String) - Method in class ai.djl.modality.nlp.DefaultVocabulary
Check if the vocabulary contains a token.
contains(String) - Method in interface ai.djl.modality.nlp.Vocabulary
Check if the vocabulary contains a token.
contains(String) - Method in class ai.djl.ndarray.NDList
Returns true if this NDList contains an NDArray with the specified name.
contains(Version) - Method in class ai.djl.repository.VersionRange
Returns true if a version falls within this range.
contains(Artifact) - Method in class ai.djl.repository.VersionRange
Returns true if the artifact's version falls within this range.
contentEquals(Number) - Method in interface ai.djl.ndarray.NDArray
Returns true if all elements in this NDArray are equal to the Number.
contentEquals(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns true if all elements in this NDArray are equal to the other NDArray.
contentEquals(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns true if all elements in this NDArray are equal to the Number.
contentEquals(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns true if all elements in this NDArray are equal to the other NDArray.
contentEquals(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns true if all elements in NDArray a are equal to NDArray b.
contentEquals(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns true if all elements in NDArray a are equal to NDArray b.
Conv1d - Class in ai.djl.nn.convolutional
A Conv1d layer works similar to Convolution layer with the exception of the number of dimension it operates on being only one, which is LayoutType.WIDTH.
conv1d(NDArray, NDArray) - Static method in class ai.djl.nn.convolutional.Conv1d
Applies 1D convolution over an input signal composed of several input planes.
conv1d(NDArray, NDArray, NDArray) - Static method in class ai.djl.nn.convolutional.Conv1d
Applies 1D convolution over an input signal composed of several input planes.
conv1d(NDArray, NDArray, NDArray, Shape) - Static method in class ai.djl.nn.convolutional.Conv1d
Applies 1D convolution over an input signal composed of several input planes.
conv1d(NDArray, NDArray, NDArray, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv1d
Applies 1D convolution over an input signal composed of several input planes.
conv1d(NDArray, NDArray, NDArray, Shape, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv1d
Applies 1D convolution over an input signal composed of several input planes.
conv1d(NDArray, NDArray, NDArray, Shape, Shape, Shape, int) - Static method in class ai.djl.nn.convolutional.Conv1d
Applies 1D convolution over an input signal composed of several input planes.
Conv1d.Builder - Class in ai.djl.nn.convolutional
The Builder to construct a Conv1d type of Block.
Conv1dTranspose - Class in ai.djl.nn.convolutional
A Conv1dTranspose layer works similar to Deconvolution layer with the exception of the number of dimension it operates on being only one, which is LayoutType.WIDTH.
conv1dTranspose(NDArray, NDArray) - Static method in class ai.djl.nn.convolutional.Conv1dTranspose
Applies 1D deconvolution over an input signal composed of several input planes.
conv1dTranspose(NDArray, NDArray, NDArray) - Static method in class ai.djl.nn.convolutional.Conv1dTranspose
Applies 1D deconvolution over an input signal composed of several input planes.
conv1dTranspose(NDArray, NDArray, NDArray, Shape) - Static method in class ai.djl.nn.convolutional.Conv1dTranspose
Applies 1D deconvolution over an input signal composed of several input planes.
conv1dTranspose(NDArray, NDArray, NDArray, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv1dTranspose
Applies 1D deconvolution over an input signal composed of several input planes.
conv1dTranspose(NDArray, NDArray, NDArray, Shape, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv1dTranspose
Applies 1D deconvolution over an input signal composed of several input planes.
conv1dTranspose(NDArray, NDArray, NDArray, Shape, Shape, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv1dTranspose
Applies 1D deconvolution over an input signal composed of several input planes.
conv1dTranspose(NDArray, NDArray, NDArray, Shape, Shape, Shape, Shape, int) - Static method in class ai.djl.nn.convolutional.Conv1dTranspose
Applies 1D convolution over an input signal composed of several input planes.
Conv1dTranspose.Builder - Class in ai.djl.nn.convolutional
The Builder to construct a Conv1dTranspose type of Block.
Conv2d - Class in ai.djl.nn.convolutional
Being the pioneer of convolution layers, Conv2d layer works on two dimensions of input, LayoutType.WIDTH and LayoutType.HEIGHT as usually a Conv2d layer is used to process data with two spatial dimensions, namely image.
conv2d(NDArray, NDArray) - Static method in class ai.djl.nn.convolutional.Conv2d
Applies 2D convolution over an input signal composed of several input planes.
conv2d(NDArray, NDArray, NDArray) - Static method in class ai.djl.nn.convolutional.Conv2d
Applies 2D convolution over an input signal composed of several input planes.
conv2d(NDArray, NDArray, NDArray, Shape) - Static method in class ai.djl.nn.convolutional.Conv2d
Applies 2D convolution over an input signal composed of several input planes.
conv2d(NDArray, NDArray, NDArray, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv2d
Applies 2D convolution over an input signal composed of several input planes.
conv2d(NDArray, NDArray, NDArray, Shape, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv2d
Applies 2D convolution over an input signal composed of several input planes.
conv2d(NDArray, NDArray, NDArray, Shape, Shape, Shape, int) - Static method in class ai.djl.nn.convolutional.Conv2d
Applies 2D convolution over an input signal composed of several input planes.
Conv2d.Builder - Class in ai.djl.nn.convolutional
The Builder to construct a Conv2d type of Block.
Conv2dTranspose - Class in ai.djl.nn.convolutional
The input to a Conv2dTranspose is an NDList with a single 4-D NDArray.
conv2dTranspose(NDArray, NDArray) - Static method in class ai.djl.nn.convolutional.Conv2dTranspose
Applies 2D deconvolution over an input signal composed of several input planes.
conv2dTranspose(NDArray, NDArray, NDArray) - Static method in class ai.djl.nn.convolutional.Conv2dTranspose
Applies 2D deconvolution over an input signal composed of several input planes.
conv2dTranspose(NDArray, NDArray, NDArray, Shape) - Static method in class ai.djl.nn.convolutional.Conv2dTranspose
Applies 2D deconvolution over an input signal composed of several input planes.
conv2dTranspose(NDArray, NDArray, NDArray, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv2dTranspose
Applies 2D deconvolution over an input signal composed of several input planes.
conv2dTranspose(NDArray, NDArray, NDArray, Shape, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv2dTranspose
Applies 2D deconvolution over an input signal composed of several input planes.
conv2dTranspose(NDArray, NDArray, NDArray, Shape, Shape, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv2dTranspose
Applies 2D deconvolution over an input signal composed of several input planes.
conv2dTranspose(NDArray, NDArray, NDArray, Shape, Shape, Shape, Shape, int) - Static method in class ai.djl.nn.convolutional.Conv2dTranspose
Applies 2D deconvolution over an input signal composed of several input planes.
Conv2dTranspose.Builder - Class in ai.djl.nn.convolutional
The Builder to construct a Conv2dTranspose type of Block.
Conv3d - Class in ai.djl.nn.convolutional
Conv3d layer behaves just as Convolution does, with the distinction being it operates of 3-dimensional data such as medical images or video data.
conv3d(NDArray, NDArray) - Static method in class ai.djl.nn.convolutional.Conv3d
Applies 3D convolution over an input signal composed of several input planes.
conv3d(NDArray, NDArray, NDArray) - Static method in class ai.djl.nn.convolutional.Conv3d
Applies 3D convolution over an input signal composed of several input planes.
conv3d(NDArray, NDArray, NDArray, Shape) - Static method in class ai.djl.nn.convolutional.Conv3d
Applies 3D convolution over an input signal composed of several input planes.
conv3d(NDArray, NDArray, NDArray, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv3d
Applies 3D convolution over an input signal composed of several input planes.
conv3d(NDArray, NDArray, NDArray, Shape, Shape, Shape) - Static method in class ai.djl.nn.convolutional.Conv3d
Applies 3D convolution over an input signal composed of several input planes.
conv3d(NDArray, NDArray, NDArray, Shape, Shape, Shape, int) - Static method in class ai.djl.nn.convolutional.Conv3d
Applies 3D convolution over an input signal composed of several input planes.
Conv3d.Builder - Class in ai.djl.nn.convolutional
The Builder to construct a Conv3d type of Block.
Convolution - Class in ai.djl.nn.convolutional
A convolution layer does a dot product calculation on each channel of \(k\)-channel input data by specified number of filters, each containing \(k\) kernels for calculating each channel in the input data and then summed per filter, hence the number of filters denote the number of output channels of a convolution layer.
Convolution(Convolution.ConvolutionBuilder<?>) - Constructor for class ai.djl.nn.convolutional.Convolution
Creates a Convolution object.
Convolution.ConvolutionBuilder<T extends Convolution.ConvolutionBuilder> - Class in ai.djl.nn.convolutional
A builder that can build any Convolution block.
ConvolutionBuilder() - Constructor for class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
 
copyBuffer(Buffer, ByteBuffer) - Static method in class ai.djl.ndarray.BaseNDManager
Copies data from the source Buffer to the target ByteBuffer.
copyTo(NDArray) - Method in interface ai.djl.ndarray.NDArray
Deep-copies the current NDArray to the one passed in.
copyTo(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Deep-copies the current NDArray to the one passed in.
correctInstances - Variable in class ai.djl.training.evaluator.AbstractAccuracy
 
cos() - Method in interface ai.djl.ndarray.NDArray
Returns the trigonometric cosine of this NDArray element-wise.
cos() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the trigonometric cosine of this NDArray element-wise.
cosh() - Method in interface ai.djl.ndarray.NDArray
Returns the hyperbolic cosine of this NDArray element-wise.
cosh() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the hyperbolic cosine of this NDArray element-wise.
cosine() - Static method in interface ai.djl.training.tracker.Tracker
Returns a new instance of CosineTracker.Builder that can build an CosineTracker.
CosineTracker - Class in ai.djl.training.tracker
CosineTracker is an implementation of Tracker which is updated by taking sections of a cosine curve to smoothly reduce learning rate until a specified step and base learning rate.
CosineTracker(CosineTracker.Builder) - Constructor for class ai.djl.training.tracker.CosineTracker
Creates a new instance of CosineTracker.
CosineTracker.Builder - Class in ai.djl.training.tracker
The Builder to construct an CosineTracker object.
countNonzero() - Method in interface ai.djl.ndarray.NDArray
Counts the number of non-zero values in this NDArray.
countNonzero(int) - Method in interface ai.djl.ndarray.NDArray
Counts the number of non-zero values in this NDArray along a given axis.
cpu() - Static method in class ai.djl.Device
Returns the default CPU Device.
CPU - Static variable in interface ai.djl.Device.Type
 
create(String[], Charset, Shape) - Method in class ai.djl.ndarray.BaseNDManager
Creates a String NDArray based on the provided shape.
create(Shape, DataType) - Method in class ai.djl.ndarray.BaseNDManager
Creates an uninitialized instance of NDArray with specified Shape, and DataType.
create(Shape) - Method in interface ai.djl.ndarray.NDManager
Creates an uninitialized instance of DataType.FLOAT32 NDArray with specified Shape.
create(Number) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a scalar NDArray.
create(float) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a scalar NDArray.
create(int) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a scalar NDArray.
create(double) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a scalar NDArray.
create(long) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a scalar NDArray.
create(byte) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a scalar NDArray.
create(boolean) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a scalar NDArray.
create(String) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a scalar NDArray.
create(String[]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes 1D NDArray.
create(String[], Charset) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes 1D NDArray.
create(String[], Shape) - Method in interface ai.djl.ndarray.NDManager
Creates a String NDArray based on the provided shape.
create(String[], Charset, Shape) - Method in interface ai.djl.ndarray.NDManager
Creates a String NDArray based on the provided shape.
create(float[]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 1D NDArray.
create(int[]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 1D NDArray.
create(double[]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 1D NDArray.
create(long[]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 1D NDArray.
create(byte[]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 1D NDArray.
create(boolean[]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 1D NDArray.
create(float[][]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 2D NDArray.
create(int[][]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 2D NDArray.
create(double[][]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 2D NDArray.
create(long[][]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 2-D NDArray.
create(byte[][]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 2-D NDArray.
create(boolean[][]) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a 2-D NDArray.
create(Buffer, Shape) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes a NDArray with specified Shape.
create(Shape, DataType) - Method in interface ai.djl.ndarray.NDManager
Creates an uninitialized instance of NDArray with specified Shape, and DataType.
create(Buffer, Shape, DataType) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes an instance of NDArray with specified Shape and DataType.
create(float[], Shape) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes an instance of NDArray with specified Shape and float array.
create(int[], Shape) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes an instance of NDArray with specified Shape and int array.
create(double[], Shape) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes an instance of NDArray with specified Shape and double array.
create(long[], Shape) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes an instance of NDArray with specified Shape and long array.
create(byte[], Shape) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes an instance of NDArray with specified Shape and byte array.
create(boolean[], Shape) - Method in interface ai.djl.ndarray.NDManager
Creates and initializes an instance of NDArray with specified Shape and boolean array.
create(Shape, DataType, Device) - Method in interface ai.djl.ndarray.NDManager
Creates an uninitialized instance of NDArray with specified Shape, DataType and Device.
createAttentionMaskFromInputMask(NDArray, NDArray) - Static method in class ai.djl.nn.transformer.BertBlock
Creates a 3D attention mask from a 2D tensor mask.
createCoo(Buffer, long[][], Shape) - Method in class ai.djl.ndarray.BaseNDManager
Creates a Coordinate Format (COO) Matrix.
createCoo(Buffer, long[][], Shape) - Method in interface ai.djl.ndarray.NDManager
Creates a Coordinate Format (COO) Matrix.
createCSR(Buffer, long[], long[], Shape) - Method in class ai.djl.ndarray.BaseNDManager
Creates a Compressed Sparse Row Storage (CSR) Format Matrix.
createCSR(float[], long[], long[], Shape, Device) - Method in interface ai.djl.ndarray.NDManager
Creates a Compressed Sparse Row Storage (CSR) Format Matrix.
createCSR(Buffer, long[], long[], Shape, Device) - Method in interface ai.djl.ndarray.NDManager
Creates a Compressed Sparse Row Storage (CSR) Format Matrix.
createCSR(Buffer, long[], long[], Shape) - Method in interface ai.djl.ndarray.NDManager
Creates a Compressed Sparse Row Storage (CSR) Format Matrix.
createModel(Path, String, Device, Block, Map<String, Object>, String) - Method in class ai.djl.repository.zoo.BaseModelLoader
 
createRowSparse(Buffer, Shape, long[], Shape) - Method in class ai.djl.ndarray.BaseNDManager
Stores the matrix in row sparse format.
createRowSparse(Buffer, Shape, long[], Shape, Device) - Method in interface ai.djl.ndarray.NDManager
Stores the matrix in row sparse format.
createRowSparse(Buffer, Shape, long[], Shape) - Method in interface ai.djl.ndarray.NDManager
Stores the matrix in row sparse format.
Criteria<I,O> - Class in ai.djl.repository.zoo
The Criteria class contains search criteria to look up a ZooModel.
Criteria.Builder<I,O> - Class in ai.djl.repository.zoo
A Builder to construct a Criteria.
Crop - Class in ai.djl.modality.cv.transform
A Transform that crops the image to a given location and size.
Crop(int, int, int, int) - Constructor for class ai.djl.modality.cv.transform.Crop
Creates a CenterCrop Transform.
crop(NDArray, int, int, int, int) - Static method in class ai.djl.modality.cv.util.NDImageUtils
Crops an image with a given location and size.
CUDA - Static variable in class ai.djl.engine.StandardCapabilities
 
CUDNN - Static variable in class ai.djl.engine.StandardCapabilities
 
cumSum() - Method in interface ai.djl.ndarray.NDArray
Returns the cumulative sum of the elements in the flattened NDArray.
cumSum(int) - Method in interface ai.djl.ndarray.NDArray
Return the cumulative sum of the elements along a given axis.
cumSum() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the cumulative sum of the elements in the flattened NDArray.
cumSum(int) - Method in class ai.djl.ndarray.NDArrayAdapter
Return the cumulative sum of the elements along a given axis.

D

data - Variable in class ai.djl.training.dataset.ArrayDataset
 
dataBatchifier - Variable in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
 
dataBatchifier - Variable in class ai.djl.training.dataset.RandomAccessDataset
 
DataDesc - Class in ai.djl.ndarray.types
A data descriptor class that encapsulates information of a NDArray.
DataDesc(Shape) - Constructor for class ai.djl.ndarray.types.DataDesc
Constructs and initializes a DataDesc with specified Shape.
DataDesc(Shape, String) - Constructor for class ai.djl.ndarray.types.DataDesc
Constructs and initializes a DataDesc with specified Shape and name.
DataDesc(Shape, DataType) - Constructor for class ai.djl.ndarray.types.DataDesc
Constructs and initializes a DataDesc with specified Shape and DataType.
DataDesc(Shape, DataType, String) - Constructor for class ai.djl.ndarray.types.DataDesc
Constructs and initializes a DataDesc with specified Shape, DataType, name, Device and SparseFormat.
DataIterable - Class in ai.djl.training.dataset
DataIterable is a data loader that combines Dataset, Batchifier, Pipeline, and Sampler to provide an iterable over the given RandomAccessDataset.
DataIterable(RandomAccessDataset, NDManager, Sampler, Batchifier, Batchifier, Pipeline, Pipeline, ExecutorService, int, Device) - Constructor for class ai.djl.training.dataset.DataIterable
Creates a new instance of DataIterable with the given parameters.
dataset(Repository, Application, String, String, String) - Static method in class ai.djl.repository.MRL
Creates a dataset MRL with specified application.
dataset(Application, String, String) - Method in interface ai.djl.repository.Repository
Creates a dataset MRL with specified application.
dataset(Application, String, String, String) - Method in interface ai.djl.repository.Repository
Creates a dataset MRL with specified application.
Dataset - Interface in ai.djl.training.dataset
An interface to represent a set of sample data/label pairs to train a model.
Dataset.Usage - Enum in ai.djl.training.dataset
An enum that indicates the mode - training, test or validation.
dataType - Variable in class ai.djl.BaseModel
 
dataType - Variable in class ai.djl.ndarray.NDArrayAdapter
 
DataType - Enum in ai.djl.ndarray.types
An enum representing the underlying NDArray's data type.
DataType.Format - Enum in ai.djl.ndarray.types
The general data type format categories.
debugDump(int) - Method in class ai.djl.ndarray.BaseNDManager
Prints information about this NDManager and all sub-managers to the console.
debugEnvironment() - Static method in class ai.djl.engine.Engine
Prints debug information about the environment for debugging environment issues.
decode(byte[]) - Method in class ai.djl.modality.nlp.embedding.TrainableWordEmbedding
Decodes the given byte array into an object of input parameter type.
decode(NDManager, byte[]) - Static method in interface ai.djl.ndarray.NDArray
Decodes NDArray from bytes.
decode(NDManager, byte[]) - Static method in class ai.djl.ndarray.NDList
Decodes NDList from byte array.
decode(NDManager, InputStream) - Static method in class ai.djl.ndarray.NDList
Decodes NDList from InputStream.
decode(byte[]) - Method in interface ai.djl.ndarray.NDManager
Decodes NDArray through byte array.
decode(InputStream) - Method in interface ai.djl.ndarray.NDManager
Decodes NDArray through DataInputStream.
decode(DataInputStream) - Static method in class ai.djl.ndarray.types.Shape
Decodes the data in the given DataInputStream and converts it into the corresponding Shape object.
decode(byte[]) - Method in interface ai.djl.nn.core.AbstractIndexedEmbedding
Decodes the given byte array into an object of input parameter type.
decode(byte[]) - Method in class ai.djl.nn.core.ConstantEmbedding
Decodes the given byte array into an object of input parameter type.
decode(byte[]) - Method in class ai.djl.nn.core.Embedding.DefaultEmbedding
Decodes the given byte array into an object of input parameter type.
decode(byte[]) - Method in class ai.djl.nn.core.Embedding.DefaultItem
Decodes the given byte array into an object of input parameter type.
Decoder - Class in ai.djl.modality.nlp
Decoder is an abstract block that be can used as decoder in encoder-decoder architecture.
Decoder(byte, Block) - Constructor for class ai.djl.modality.nlp.Decoder
Constructs a new instance of Decoder with the given block.
decoder - Variable in class ai.djl.modality.nlp.EncoderDecoder
 
Deconvolution - Class in ai.djl.nn.convolutional
Transposed convolution, also named fractionally-strided convolution Dumoulin & Visin or deconvolution Long et al., 2015, serves this purpose.
Deconvolution(Deconvolution.DeconvolutionBuilder<?>) - Constructor for class ai.djl.nn.convolutional.Deconvolution
Creates a Deconvolution object.
Deconvolution.DeconvolutionBuilder<T extends Deconvolution.DeconvolutionBuilder> - Class in ai.djl.nn.convolutional
A builder that can build any Deconvolution block.
DeconvolutionBuilder() - Constructor for class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
 
DEFAULT_FORM - Static variable in class ai.djl.modality.nlp.preprocess.UnicodeNormalizer
 
defaultDevice() - Method in class ai.djl.engine.Engine
Returns the engine's default Device.
defaultDevice() - Method in class ai.djl.ndarray.BaseNDManager
Returns the default context used in Engine.
defaultDevice() - Method in interface ai.djl.ndarray.NDManager
Returns the default context used in Engine.
DefaultEmbedding() - Constructor for class ai.djl.nn.core.Embedding.DefaultEmbedding
 
defaultFactory - Variable in class ai.djl.repository.zoo.BaseModelLoader
 
defaultItem - Variable in class ai.djl.nn.core.Embedding.BaseBuilder
 
DefaultItem(T) - Constructor for class ai.djl.nn.core.Embedding.DefaultItem
 
DefaultModelZoo - Class in ai.djl.repository.zoo
A ModelZoo that contains models in specified locations.
DefaultModelZoo() - Constructor for class ai.djl.repository.zoo.DefaultModelZoo
Constructs a new LocalModelZoo instance.
DefaultModelZoo(String) - Constructor for class ai.djl.repository.zoo.DefaultModelZoo
Constructs a new LocalModelZoo instance from the given search locations.
DefaultTrainingConfig - Class in ai.djl.training
DefaultTrainingConfig is an implementation of the TrainingConfig interface.
DefaultTrainingConfig(Loss) - Constructor for class ai.djl.training.DefaultTrainingConfig
Creates an instance of DefaultTrainingConfig with the given Loss.
DefaultTranslatorFactory - Class in ai.djl.translate
A default implementation of TranslatorFactory.
DefaultTranslatorFactory() - Constructor for class ai.djl.translate.DefaultTranslatorFactory
 
DefaultVocabulary - Class in ai.djl.modality.nlp
The default implementation of Vocabulary.
DefaultVocabulary(List<String>) - Constructor for class ai.djl.modality.nlp.DefaultVocabulary
Creates a DefaultVocabulary object with the given list of tokens.
DefaultVocabulary(DefaultVocabulary.Builder) - Constructor for class ai.djl.modality.nlp.DefaultVocabulary
Creates a DefaultVocabulary object with a DefaultVocabulary.Builder.
DefaultVocabulary.Builder - Class in ai.djl.modality.nlp
Builder class that is used to build the DefaultVocabulary.
DefaultZooProvider - Class in ai.djl.repository.zoo
An ZooProvider implementation can load models from specified locations.
DefaultZooProvider() - Constructor for class ai.djl.repository.zoo.DefaultZooProvider
 
describeInput() - Method in class ai.djl.BaseModel
Returns the input descriptor of the model.
describeInput() - Method in class ai.djl.modality.nlp.EncoderDecoder
Returns a PairList of input names, and shapes.
describeInput() - Method in interface ai.djl.Model
Returns the input descriptor of the model.
describeInput() - Method in class ai.djl.nn.AbstractBlock
Returns a PairList of input names, and shapes.
describeInput() - Method in interface ai.djl.nn.Block
Returns a PairList of input names, and shapes.
describeInput() - Method in class ai.djl.nn.core.Linear
Returns a PairList of input names, and shapes.
describeInput() - Method in class ai.djl.repository.zoo.ZooModel
Returns the input descriptor of the model.
describeOutput() - Method in class ai.djl.BaseModel
Returns the output descriptor of the model.
describeOutput() - Method in interface ai.djl.Model
Returns the output descriptor of the model.
describeOutput() - Method in interface ai.djl.nn.SymbolBlock
Returns a PairList of output names and shapes stored in model file.
describeOutput() - Method in class ai.djl.repository.zoo.ZooModel
Returns the output descriptor of the model.
detach() - Method in class ai.djl.ndarray.NDList
Detaches the NDResource from current NDManager's lifecycle.
detach() - Method in interface ai.djl.ndarray.NDResource
Detaches the NDResource from current NDManager's lifecycle.
detach() - Method in class ai.djl.training.util.MinMaxScaler
Detaches this MinMaxScaler fitted value from current NDManager's lifecycle.
detachInternal(String) - Method in class ai.djl.ndarray.BaseNDManager
Detaches a NDArray from this NDManager's lifecycle.
detachInternal(String) - Method in interface ai.djl.ndarray.NDManager
Detaches a NDArray from this NDManager's lifecycle.
DetectedObject(String, double, BoundingBox) - Constructor for class ai.djl.modality.cv.output.DetectedObjects.DetectedObject
Constructs a bounding box with the given data.
DetectedObjects - Class in ai.djl.modality.cv.output
A class representing the detected objects results for a single image in an Application.CV.OBJECT_DETECTION case.
DetectedObjects(List<String>, List<Double>, List<BoundingBox>) - Constructor for class ai.djl.modality.cv.output.DetectedObjects
Constructs a DetectedObjects, usually during post-processing.
DetectedObjects.DetectedObject - Class in ai.djl.modality.cv.output
A DetectedObject represents a single potential detected Object for an image.
detection(NDList) - Method in class ai.djl.modality.cv.MultiBoxDetection
Converts multi-box detection predictions.
Device - Class in ai.djl
The Device class provides the specified assignment for CPU/GPU processing on the NDArray.
device - Variable in class ai.djl.ndarray.BaseNDManager
 
device - Variable in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
 
device - Variable in class ai.djl.training.dataset.RandomAccessDataset
 
Device.Type - Interface in ai.djl
Contains device type string constants.
DeviceType - Interface in ai.djl
DeviceType is a class used to map the Device name to their corresponding type number.
dilation - Variable in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
 
dilation - Variable in class ai.djl.nn.convolutional.Convolution
 
dilation - Variable in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
 
dilation - Variable in class ai.djl.nn.convolutional.Deconvolution
 
dimension() - Method in class ai.djl.ndarray.types.Shape
Returns the number of dimensions of this Shape.
div(Number) - Method in interface ai.djl.ndarray.NDArray
Divides this NDArray by a number element-wise.
div(NDArray) - Method in interface ai.djl.ndarray.NDArray
Divides this NDArray by the other NDArray element-wise.
div(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Divides this NDArray by a number element-wise.
div(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Divides this NDArray by the other NDArray element-wise.
div(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Divides the NDArray by a number element-wise.
div(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Divides a number by a NDArray element-wise.
div(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Divides a NDArray by a NDArray element-wise.
DivergenceCheckTrainingListener - Class in ai.djl.training.listener
TrainingListener that gives early warning if your training has failed by divergence.
DivergenceCheckTrainingListener() - Constructor for class ai.djl.training.listener.DivergenceCheckTrainingListener
 
divi(Number) - Method in interface ai.djl.ndarray.NDArray
Divides this NDArray by a number element-wise in place.
divi(NDArray) - Method in interface ai.djl.ndarray.NDArray
Divides this NDArray by the other NDArray element-wise in place.
divi(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Divides this NDArray by a number element-wise in place.
divi(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Divides this NDArray by the other NDArray element-wise in place.
divi(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Divides a number by a NDArray element-wise in place.
divi(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Divides a number by a NDArray element-wise.
divi(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Divides a NDArray by a NDArray element-wise.
dot(NDArray) - Method in interface ai.djl.ndarray.NDArray
Dot product of this NDArray and the other NDArray.
dot(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Dot product of this NDArray and the other NDArray.
dot(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Dot product of NDArray a and NDArray b.
download(Path, URI, Artifact.Item, Progress) - Method in class ai.djl.repository.AbstractRepository
 
download(Path, URI, Artifact.Item, Progress) - Method in class ai.djl.repository.JarRepository
download(Path, URI, Artifact.Item, Progress) - Method in class ai.djl.repository.SimpleRepository
download(Path, URI, Artifact.Item, Progress) - Method in class ai.djl.repository.SimpleUrlRepository
download(String, String) - Static method in class ai.djl.training.util.DownloadUtils
Downloads a file from specified url.
download(String, String, Progress) - Static method in class ai.djl.training.util.DownloadUtils
Downloads a file from specified url.
download(URL, Path, Progress) - Static method in class ai.djl.training.util.DownloadUtils
Downloads a file from specified url.
DownloadUtils - Class in ai.djl.training.util
A utility class downloads the file from specified url.
drawBoundingBoxes(DetectedObjects) - Method in interface ai.djl.modality.cv.Image
Draws the bounding boxes on the image.
drawJoints(Joints) - Method in interface ai.djl.modality.cv.Image
Draws all joints of a body on an image.
Dropout - Class in ai.djl.nn.norm
A dropout layer benefits a network by allowing some units (neurons), and hence their respective connections, of a network to be randomly and temporarily removed by setting its value to 0 only during training by specified probability \(p\), usually set to 0.5.
dropout(NDArray) - Static method in class ai.djl.nn.norm.Dropout
Applies Dropout to the input.
dropout(NDArray, float) - Static method in class ai.djl.nn.norm.Dropout
Applies Dropout to the input.
dropout(NDArray, float, boolean) - Static method in class ai.djl.nn.norm.Dropout
Applies Dropout to the input.
Dropout.Builder - Class in ai.djl.nn.norm
The Builder to construct a Dropout type of Block.
dropRate - Variable in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
 
dropRate - Variable in class ai.djl.nn.recurrent.RecurrentBlock
 
dumpMemoryInfo(Metrics, String) - Static method in class ai.djl.training.listener.MemoryTrainingListener
Dump memory metrics into log directory.
duplicate() - Method in interface ai.djl.modality.cv.Image
Gets a deep copy of the original image.
duplicate() - Method in interface ai.djl.ndarray.NDArray
Returns a copy of this NDArray.

E

EasyHpo - Class in ai.djl.training.hyperparameter
Helper for easy training with hyperparameters.
EasyHpo() - Constructor for class ai.djl.training.hyperparameter.EasyHpo
 
EasyTrain - Class in ai.djl.training
Helper for easy training of a whole model, a trainining batch, or a validation batch.
ElasticNetWeightDecay - Class in ai.djl.training.loss
ElasticWeightDecay calculates L1+L2 penalty of a set of parameters.
ElasticNetWeightDecay(NDList) - Constructor for class ai.djl.training.loss.ElasticNetWeightDecay
Calculates Elastic Net weight decay for regularization.
ElasticNetWeightDecay(String, NDList) - Constructor for class ai.djl.training.loss.ElasticNetWeightDecay
Calculates Elastic Net weight decay for regularization.
ElasticNetWeightDecay(String, NDList, float) - Constructor for class ai.djl.training.loss.ElasticNetWeightDecay
Calculates Elastic Net weight decay for regularization.
ElasticNetWeightDecay(String, NDList, float, float) - Constructor for class ai.djl.training.loss.ElasticNetWeightDecay
Calculates Elastic Net weight decay for regularization.
elasticNetWeightedDecay(NDList) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of ElasticNetWeightDecay with default weight and name.
elasticNetWeightedDecay(String, NDList) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of ElasticNetWeightDecay with default weight.
elasticNetWeightedDecay(String, float, NDList) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of ElasticNetWeightDecay.
elasticNetWeightedDecay(String, float, float, NDList) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of ElasticNetWeightDecay.
elu(NDArray, float) - Static method in class ai.djl.nn.Activation
Applies ELU activation on the input NDArray.
elu(NDList, float) - Static method in class ai.djl.nn.Activation
Applies ELU(Exponential Linear Unit) activation on the input singleton NDList.
eluBlock(float) - Static method in class ai.djl.nn.Activation
Creates a LambdaBlock that applies the ELU activation function in its forward function.
embed(String) - Method in class ai.djl.modality.nlp.embedding.TrainableWordEmbedding
Embeds an item.
embed(NDManager, T[]) - Method in interface ai.djl.nn.core.AbstractEmbedding
Embeds an array of items.
embed(T) - Method in interface ai.djl.nn.core.AbstractIndexedEmbedding
Embeds an item.
embed(Object) - Method in class ai.djl.nn.core.ConstantEmbedding
Embeds an item.
embed(NDManager, Object[]) - Method in class ai.djl.nn.core.ConstantEmbedding
Embeds an array of items.
embed(NDManager, T[]) - Method in class ai.djl.nn.core.Embedding.DefaultEmbedding
Embeds an array of items.
embed(T) - Method in class ai.djl.nn.core.Embedding.DefaultEmbedding
Embeds an item.
embed(NDManager, T[]) - Method in class ai.djl.nn.core.Embedding.DefaultItem
Embeds an array of items.
embed(T) - Method in class ai.djl.nn.core.Embedding.DefaultItem
Embeds an item.
embed(NDManager, T[]) - Method in class ai.djl.nn.core.Embedding
Embeds an array of items.
embedding - Variable in class ai.djl.nn.core.ConstantEmbedding
 
Embedding<T> - Class in ai.djl.nn.core
An Embedding block map a collection of items to 1-Dimensional representative NDArrays.
Embedding(Embedding.BaseBuilder<T, ?>) - Constructor for class ai.djl.nn.core.Embedding
 
Embedding(NDArray) - Constructor for class ai.djl.nn.core.Embedding
Constructs a pretrained embedding.
Embedding(NDArray, SparseFormat) - Constructor for class ai.djl.nn.core.Embedding
Constructs a pretrained embedding.
embedding - Variable in class ai.djl.nn.core.Embedding
 
embedding(NDArray, NDArray, SparseFormat) - Static method in class ai.djl.nn.core.Embedding
A simple lookup table that looks up embeddings in a fixed dictionary and size.
Embedding.BaseBuilder<T,B extends Embedding.BaseBuilder<T,B>> - Class in ai.djl.nn.core
The Builder to construct a Embedding type of Block.
Embedding.DefaultEmbedding - Class in ai.djl.nn.core
 
Embedding.DefaultItem - Class in ai.djl.nn.core
 
EmbeddingException - Exception in ai.djl.modality.nlp.embedding
Thrown to indicate that there was some error while loading embeddings.
EmbeddingException(String) - Constructor for exception ai.djl.modality.nlp.embedding.EmbeddingException
Constructs a new exception with the specified detail message.
EmbeddingException(String, Throwable) - Constructor for exception ai.djl.modality.nlp.embedding.EmbeddingException
Constructs a new exception with the specified detail message and cause.
EmbeddingException(Throwable) - Constructor for exception ai.djl.modality.nlp.embedding.EmbeddingException
Constructs a new exception with the specified cause and a detail message of (cause==null ? null : cause.toString()) which typically contains the class and detail message of cause.
embeddingSize - Variable in class ai.djl.nn.core.Embedding.BaseBuilder
 
embeddingSize - Variable in class ai.djl.nn.core.Embedding
 
embeddingType - Variable in class ai.djl.nn.core.Embedding.BaseBuilder
 
embedText(NDArray) - Method in class ai.djl.modality.nlp.embedding.ModelZooTextEmbedding
Embeds the text after preprocessed using TextEmbedding.preprocessTextToEmbed(List).
embedText(NDArray) - Method in class ai.djl.modality.nlp.embedding.SimpleTextEmbedding
Embeds the text after preprocessed using TextEmbedding.preprocessTextToEmbed(List).
embedText(NDManager, List<String>) - Method in interface ai.djl.modality.nlp.embedding.TextEmbedding
Embeds a text.
embedText(NDManager, long[]) - Method in interface ai.djl.modality.nlp.embedding.TextEmbedding
Embeds the text after preprocessed using TextEmbedding.preprocessTextToEmbed(List).
embedText(NDArray) - Method in interface ai.djl.modality.nlp.embedding.TextEmbedding
Embeds the text after preprocessed using TextEmbedding.preprocessTextToEmbed(List).
embedText(NDArray) - Method in class ai.djl.modality.nlp.embedding.TrainableTextEmbedding
Embeds the text after preprocessed using TextEmbedding.preprocessTextToEmbed(List).
embedWord(NDArray) - Method in class ai.djl.modality.nlp.embedding.TrainableWordEmbedding
Embeds the word after preprocessed using WordEmbedding.preprocessWordToEmbed(String).
embedWord(NDManager, String) - Method in interface ai.djl.modality.nlp.embedding.WordEmbedding
Embeds a word.
embedWord(NDManager, long) - Method in interface ai.djl.modality.nlp.embedding.WordEmbedding
Embeds the word after preprocessed using WordEmbedding.preprocessWordToEmbed(String).
embedWord(NDArray) - Method in interface ai.djl.modality.nlp.embedding.WordEmbedding
Embeds the word after preprocessed using WordEmbedding.preprocessWordToEmbed(String).
encode(String, String) - Method in class ai.djl.modality.nlp.bert.BertTokenizer
Encodes questions and paragraph sentences.
encode(String, String, int) - Method in class ai.djl.modality.nlp.bert.BertTokenizer
Encodes questions and paragraph sentences with max length.
encode(String) - Method in class ai.djl.modality.nlp.embedding.TrainableWordEmbedding
Encodes an object of input type into a byte array.
encode() - Method in interface ai.djl.ndarray.NDArray
Encodes NDArray to byte array.
encode() - Method in class ai.djl.ndarray.NDList
Encodes the NDList to byte array.
encode(OutputStream) - Method in class ai.djl.ndarray.NDList
Writes the encoded NDList to OutputStream.
encode(OutputStream, boolean) - Method in class ai.djl.ndarray.NDList
Writes the encoded NDList to OutputStream.
encode(T) - Method in interface ai.djl.nn.core.AbstractIndexedEmbedding
Encodes an object of input type into a byte array.
encode(Object) - Method in class ai.djl.nn.core.ConstantEmbedding
Encodes an object of input type into a byte array.
encode(T) - Method in class ai.djl.nn.core.Embedding.DefaultEmbedding
Encodes an object of input type into a byte array.
encode(T) - Method in class ai.djl.nn.core.Embedding.DefaultItem
Encodes an object of input type into a byte array.
Encoder - Class in ai.djl.modality.nlp
Encoder is an abstract block that be can used as encoder in encoder-decoder architecture.
Encoder(byte, Block) - Constructor for class ai.djl.modality.nlp.Encoder
Constructs a new instance of Encoder with the given block.
encoder - Variable in class ai.djl.modality.nlp.EncoderDecoder
 
EncoderDecoder - Class in ai.djl.modality.nlp
EncoderDecoder is a general implementation of the very popular encoder-decoder architecture.
EncoderDecoder(Encoder, Decoder) - Constructor for class ai.djl.modality.nlp.EncoderDecoder
Constructs a new instance of EncoderDecoder class with the given Encoder and Decoder.
end() - Method in class ai.djl.training.util.ProgressBar
Engine - Class in ai.djl.engine
The Engine interface is the base of the provided implementation for DJL.
Engine() - Constructor for class ai.djl.engine.Engine
 
EngineException - Exception in ai.djl.engine
Thrown to indicate that a native error is raised from the underlying Engine.
EngineException(String) - Constructor for exception ai.djl.engine.EngineException
Constructs a new exception with the specified detail message.
EngineException(String, Throwable) - Constructor for exception ai.djl.engine.EngineException
Constructs a new exception with the specified detail message and cause.
EngineException(Throwable) - Constructor for exception ai.djl.engine.EngineException
Constructs a new exception with the specified cause and a detail message of (cause==null ? null : cause.toString()) (which typically contains the class and detail message of cause).
EngineProvider - Interface in ai.djl.engine
The EngineProvider instance manufactures an Engine instance, which is available in the system.
EpochTrainingListener - Class in ai.djl.training.listener
EpochTrainingListener that tracks epochs.
EpochTrainingListener() - Constructor for class ai.djl.training.listener.EpochTrainingListener
 
EpsilonGreedy - Class in ai.djl.modality.rl.agent
The EpsilonGreedy is a simple exploration/excitation agent.
EpsilonGreedy(RlAgent, Tracker) - Constructor for class ai.djl.modality.rl.agent.EpsilonGreedy
Constructs an EpsilonGreedy.
eq(Number) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Equals" comparison.
eq(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Equals" comparison.
eq(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Equals" comparison.
eq(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Equals" comparison.
eq(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Equals" comparison.
eq(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Equals" comparison.
eq(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Equals" comparison.
equals(Object) - Method in class ai.djl.Application
equals(Object) - Method in class ai.djl.Device
equals(Object) - Method in class ai.djl.ndarray.types.Shape
equals(Object) - Method in class ai.djl.repository.Version
erfinv() - Method in interface ai.djl.ndarray.NDArray
Returns element-wise inverse gauss error function of the NDArray.
erfinv() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns element-wise inverse gauss error function of the NDArray.
erfinv(NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns element-wise inverse gauss error function of the input NDArray.
evaluate(NDList, NDList) - Method in class ai.djl.nn.transformer.BertMaskedLanguageModelLoss
 
evaluate(NDList, NDList) - Method in class ai.djl.nn.transformer.BertNextSentenceLoss
 
evaluate(NDList, NDList) - Method in class ai.djl.training.evaluator.AbstractAccuracy
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.evaluator.BoundingBoxError
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.evaluator.Evaluator
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.loss.AbstractCompositeLoss
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.loss.ElasticNetWeightDecay
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.loss.HingeLoss
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.loss.L1Loss
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.loss.L1WeightDecay
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.loss.L2Loss
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.loss.L2WeightDecay
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.loss.MaskedSoftmaxCrossEntropyLoss
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.loss.SigmoidBinaryCrossEntropyLoss
Calculates the evaluation between the labels and the predictions.
evaluate(NDList, NDList) - Method in class ai.djl.training.loss.SoftmaxCrossEntropyLoss
Calculates the evaluation between the labels and the predictions.
evaluate(NDList) - Method in class ai.djl.training.Trainer
Evaluates function of the model once on the given input NDList.
evaluateDataset(Trainer, Dataset) - Static method in class ai.djl.training.EasyTrain
Evaluates the test dataset.
Evaluator - Class in ai.djl.training.evaluator
Base class for all Evaluators that can be used to evaluate the performance of a model.
Evaluator(String) - Constructor for class ai.djl.training.evaluator.Evaluator
Creates an evaluator with abstract update methods.
EvaluatorTrainingListener - Class in ai.djl.training.listener
TrainingListener that records evaluator results.
EvaluatorTrainingListener() - Constructor for class ai.djl.training.listener.EvaluatorTrainingListener
Constructs an EvaluatorTrainingListener that updates the training progress the default frequency.
EvaluatorTrainingListener(int) - Constructor for class ai.djl.training.listener.EvaluatorTrainingListener
Constructs an EvaluatorTrainingListener that updates the training progress the given frequency.
exp() - Method in interface ai.djl.ndarray.NDArray
Returns the exponential value of this NDArray element-wise.
exp() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the exponential value of this NDArray element-wise.
expandDims(int) - Method in interface ai.djl.ndarray.NDArray
Expands the Shape of a NDArray.
expandDims(int) - Method in class ai.djl.ndarray.NDArrayAdapter
Expands the Shape of a NDArray.
eye(int, int, int, DataType) - Method in class ai.djl.ndarray.BaseNDManager
Returns a 2-D array with ones on the diagonal and zeros elsewhere.
eye(int) - Method in interface ai.djl.ndarray.NDManager
Returns a 2-D array with ones on the diagonal and zeros elsewhere.
eye(int, int) - Method in interface ai.djl.ndarray.NDManager
Returns a 2-D array with ones on the diagonal and zeros elsewhere.
eye(int, int, int) - Method in interface ai.djl.ndarray.NDManager
Returns a 2-D array with ones on the diagonal and zeros elsewhere.
eye(int, int, int, DataType) - Method in interface ai.djl.ndarray.NDManager
Returns a 2-D array with ones on the diagonal and zeros elsewhere.
eye(int, int, int, DataType, Device) - Method in interface ai.djl.ndarray.NDManager
Returns a 2-D array with ones on the diagonal and zeros elsewhere.

F

factor() - Static method in interface ai.djl.training.tracker.Tracker
Returns a new instance of FactorTracker.Builder that can build an FactorTracker.
FactorTracker - Class in ai.djl.training.tracker
FactorTracker is an implementation of Tracker which is updated by a multiplicative factor.
FactorTracker(FactorTracker.Builder) - Constructor for class ai.djl.training.tracker.FactorTracker
Creates a new instance of FactorTracker.
FactorTracker.Builder - Class in ai.djl.training.tracker
The Builder to construct an FactorTracker object.
fallthrough - Variable in class ai.djl.nn.core.Embedding.BaseBuilder
 
fallthroughEmbedding - Variable in class ai.djl.nn.core.Embedding
 
FilenameUtils - Class in ai.djl.repository
A class containing utility methods.
FileTranslator<T> - Class in ai.djl.modality.cv.translator.wrapper
Built-in Translator that provides image pre-processing from file path.
FileTranslator(Translator<Image, T>) - Constructor for class ai.djl.modality.cv.translator.wrapper.FileTranslator
Creates a FileTranslator instance.
filterByLayoutType(Predicate<LayoutType>) - Method in class ai.djl.ndarray.types.Shape
Returns only the axes of the Shape whose layout types match the predicate.
filters - Variable in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
 
filters - Variable in class ai.djl.nn.convolutional.Convolution
 
filters - Variable in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
 
filters - Variable in class ai.djl.nn.convolutional.Deconvolution
 
finalize() - Method in class ai.djl.BaseModel
finalize() - Method in class ai.djl.inference.Predictor
finalize() - Method in class ai.djl.training.Trainer
fit(Trainer, int, Dataset, Dataset) - Static method in class ai.djl.training.EasyTrain
Runs a basic epoch training experience with a given trainer.
fit() - Method in class ai.djl.training.hyperparameter.EasyHpo
Fits the model given the implemented abstract methods.
fit(NDArray, int[]) - Method in class ai.djl.training.util.MinMaxScaler
Computes the minimum and maximum to be used for later scaling.
fit(NDArray) - Method in class ai.djl.training.util.MinMaxScaler
Computes the minimum and maximum to be used for later scaling.
fixed(float) - Static method in interface ai.djl.training.tracker.Tracker
Returns a new instance of Tracker with a fixed value.
flag - Variable in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
 
flatten() - Method in interface ai.djl.ndarray.NDArray
Flattens this NDArray into a 1-D NDArray in row-major order.
flatten() - Method in class ai.djl.ndarray.NDArrayAdapter
Flattens this NDArray into a 1-D NDArray in row-major order.
flip(int...) - Method in interface ai.djl.ndarray.NDArray
Returns the reverse order of elements in an array along the given axis.
flip(int...) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the reverse order of elements in an array along the given axis.
floatValue(Map<String, ?>, String) - Static method in class ai.djl.translate.ArgumentsUtil
Returns the float value from the arguments.
floatValue(Map<String, ?>, String, float) - Static method in class ai.djl.translate.ArgumentsUtil
Returns the float value from the arguments.
floor() - Method in interface ai.djl.ndarray.NDArray
Returns the floor of this NDArray element-wise.
floor() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the floor of this NDArray element-wise.
forward(ParameterStore, NDList, NDList, PairList<String, Object>) - Method in class ai.djl.modality.nlp.EncoderDecoder
A forward call using both training data and labels.
forward(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.AbstractBlock
Applies the operating function of the block once.
forward(ParameterStore, NDList, NDList, PairList<String, Object>) - Method in class ai.djl.nn.AbstractBlock
A forward call using both training data and labels.
forward(ParameterStore, NDList, boolean) - Method in interface ai.djl.nn.Block
Applies the operating function of the block once.
forward(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in interface ai.djl.nn.Block
Applies the operating function of the block once.
forward(ParameterStore, NDList, NDList, PairList<String, Object>) - Method in interface ai.djl.nn.Block
A forward call using both training data and labels.
forward(NDList) - Method in class ai.djl.training.Trainer
Applies the forward function of the model once on the given input NDList.
forward(NDList, NDList) - Method in class ai.djl.training.Trainer
Applies the forward function of the model once with both data and labels.
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.modality.nlp.Decoder
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.modality.nlp.embedding.TrainableTextEmbedding
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.modality.nlp.Encoder
forwardInternal(ParameterStore, NDList, NDList, PairList<String, Object>) - Method in class ai.djl.modality.nlp.Encoder
 
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.modality.nlp.EncoderDecoder
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.AbstractBlock
forwardInternal(ParameterStore, NDList, NDList, PairList<String, Object>) - Method in class ai.djl.nn.AbstractBlock
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.convolutional.Convolution
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.convolutional.Deconvolution
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.core.ConstantEmbedding
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.core.Embedding
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.core.Linear
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.core.Prelu
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.LambdaBlock
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.norm.BatchNorm
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.norm.Dropout
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.norm.LayerNorm
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.ParallelBlock
forwardInternal(ParameterStore, NDList, NDList, PairList<String, Object>) - Method in class ai.djl.nn.ParallelBlock
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.recurrent.GRU
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.recurrent.LSTM
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.recurrent.RNN
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.SequentialBlock
forwardInternal(ParameterStore, NDList, NDList, PairList<String, Object>) - Method in class ai.djl.nn.SequentialBlock
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.transformer.BertBlock
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.transformer.BertMaskedLanguageModelBlock
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.transformer.BertNextSentenceBlock
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.transformer.BertPretrainingBlock
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.transformer.IdEmbedding
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.transformer.ScaledDotProductAttentionBlock
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.nn.transformer.TransformerEncoderBlock
from(NDArray) - Method in interface ai.djl.ndarray.NDManager
Creates a new NDArray if the input NDArray is from external engine.
fromBuffer(Buffer) - Static method in enum ai.djl.ndarray.types.DataType
Returns the data type to use for a data buffer.
fromDeviceType(int) - Static method in interface ai.djl.DeviceType
Map device to its type number.
fromFile(Path) - Method in class ai.djl.modality.cv.BufferedImageFactory
Gets Image from file.
fromFile(Path) - Method in class ai.djl.modality.cv.ImageFactory
Gets Image from file.
fromImage(Object) - Method in class ai.djl.modality.cv.BufferedImageFactory
Gets Image from varies Java image types.
fromImage(Object) - Method in class ai.djl.modality.cv.ImageFactory
Gets Image from varies Java image types.
fromIndex(NDIndex, Shape) - Static method in class ai.djl.ndarray.index.full.NDIndexFullPick
Returns (if possible) the NDIndexFullPick representation of an NDIndex.
fromIndex(NDIndex, Shape) - Static method in class ai.djl.ndarray.index.full.NDIndexFullSlice
Returns (if possible) the NDIndexFullSlice representation of an NDIndex.
fromInputStream(InputStream) - Method in class ai.djl.modality.cv.BufferedImageFactory
Gets Image from InputStream.
fromInputStream(InputStream) - Method in class ai.djl.modality.cv.ImageFactory
Gets Image from InputStream.
fromNDArray(NDArray) - Method in class ai.djl.modality.cv.BufferedImageFactory
Gets Image from NDArray.
fromNDArray(NDArray) - Method in class ai.djl.modality.cv.ImageFactory
Gets Image from NDArray.
fromNumpy(String) - Static method in enum ai.djl.ndarray.types.DataType
Returns the data type from numpy value.
fromString(String) - Static method in interface ai.djl.translate.Batchifier
Returns a batchifier from a batchifier name.
fromUrl(URL) - Method in class ai.djl.modality.cv.ImageFactory
Gets Image from URL.
fromUrl(String) - Method in class ai.djl.modality.cv.ImageFactory
Gets Image from URL.
fromValue(char) - Static method in enum ai.djl.ndarray.types.LayoutType
Converts the character to the matching layout type.
fromValue(String) - Static method in enum ai.djl.ndarray.types.LayoutType
Converts each character to the matching layout type.
fromValue(int) - Static method in enum ai.djl.ndarray.types.SparseFormat
Gets the SparseFormat from it's integer value.
full(Shape, float, DataType) - Method in class ai.djl.ndarray.BaseNDManager
Return a new NDArray of given shape, filled with value.
full(Shape, int) - Method in interface ai.djl.ndarray.NDManager
Return a new NDArray of given shape, filled with value.
full(Shape, float) - Method in interface ai.djl.ndarray.NDManager
Return a new NDArray of given shape, filled with value.
full(Shape, float, DataType) - Method in interface ai.djl.ndarray.NDManager
Return a new NDArray of given shape, filled with value.
full(Shape, float, DataType, Device) - Method in interface ai.djl.ndarray.NDManager
Return a new NDArray of given shape, device, filled with value.

G

gates - Variable in class ai.djl.nn.recurrent.RecurrentBlock
 
gatherFromIndices(NDArray, NDArray) - Static method in class ai.djl.nn.transformer.BertMaskedLanguageModelBlock
Given a 3D array of shape (B, S, E) and a 2D array of shape (B, I) returns the flattened lookup result of shape (B * I * E).
gatherNd(NDArray, NDArray) - Static method in class ai.djl.nn.transformer.MissingOps
Applies the mxnet gather_nd operator.
gelu(NDArray) - Static method in class ai.djl.nn.Activation
Applies GELU(Gaussian Error Linear Unit) activation on the input NDArray.
gelu(NDList) - Static method in class ai.djl.nn.Activation
Applies GELU(Gaussian Error Linear Unit) activation on the input singleton NDList.
geluBlock() - Static method in class ai.djl.nn.Activation
Creates a LambdaBlock that applies the GELU activation function in its forward function.
generateAnchorBoxes(NDArray) - Method in class ai.djl.modality.cv.MultiBoxPrior
Generates the anchorBoxes array in the input's manager and device.
get(String) - Method in class ai.djl.modality.Classifications
Returns the result for a particular class name.
get(String) - Method in class ai.djl.modality.Input
Returns the element for the first key found in the Input.
get(int) - Method in class ai.djl.modality.Input
Returns the element at the specified position in the Input.
get(NDArray, NDIndexFullPick) - Method in class ai.djl.ndarray.index.NDArrayIndexer
Returns a subarray by picking the elements.
get(NDArray, NDIndexFullSlice) - Method in class ai.djl.ndarray.index.NDArrayIndexer
Returns a subarray at the slice.
get(NDArray, NDIndex) - Method in class ai.djl.ndarray.index.NDArrayIndexer
Returns a subarray at the given index.
get(int) - Method in class ai.djl.ndarray.index.NDIndex
Returns the index affecting the given dimension.
get(NDIndex) - Method in interface ai.djl.ndarray.NDArray
Returns a partial NDArray.
get(String, Object...) - Method in interface ai.djl.ndarray.NDArray
Returns a partial NDArray.
get(long...) - Method in interface ai.djl.ndarray.NDArray
Returns a partial NDArray.
get(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns a partial NDArray.
get(String) - Method in class ai.djl.ndarray.NDList
Returns the first occurrence of the specified element from this NDList if it is present.
get(int) - Method in class ai.djl.ndarray.types.Shape
Returns the shape in the given dimension.
get(NDManager, long) - Method in class ai.djl.training.dataset.ArrayDataset
Gets the Record for the given index from the dataset.
get(NDManager, long) - Method in class ai.djl.training.dataset.RandomAccessDataset
Gets the Record for the given index from the dataset.
get(NDManager) - Method in interface ai.djl.translate.NDArraySupplier
Gets an NDArray from the given NDManager.
getAccumulator(String) - Method in class ai.djl.training.evaluator.AbstractAccuracy
Returns the accumulated evaluator value.
getAccumulator(String) - Method in class ai.djl.training.evaluator.BoundingBoxError
Returns the accumulated evaluator value.
getAccumulator(String) - Method in class ai.djl.training.evaluator.Evaluator
Returns the accumulated evaluator value.
getAccumulator(String) - Method in class ai.djl.training.loss.AbstractCompositeLoss
Returns the accumulated evaluator value.
getAccumulator(String) - Method in class ai.djl.training.loss.Loss
Returns the accumulated evaluator value.
getAction() - Method in interface ai.djl.modality.rl.env.RlEnv.Step
Returns the action taken.
getActionSpace() - Method in interface ai.djl.modality.rl.env.RlEnv
Returns the current actions that can be taken in the environment.
getAllEngines() - Static method in class ai.djl.engine.Engine
Returns a set of engine names that are loaded.
getAlternativeEngine() - Method in class ai.djl.engine.Engine
Returns the alternative engine if available.
getAlternativeManager() - Method in class ai.djl.ndarray.BaseNDManager
 
getApplication() - Method in class ai.djl.repository.Metadata
Returns the Application.
getApplication() - Method in class ai.djl.repository.MRL
Returns the application.
getApplication() - Method in class ai.djl.repository.zoo.BaseModelLoader
Returns the application of the ModelLoader.
getApplication() - Method in class ai.djl.repository.zoo.Criteria
Returns the application of the model.
getApplication() - Method in interface ai.djl.repository.zoo.ModelLoader
Returns the application of the ModelLoader.
getArguments(Map<String, Object>) - Method in class ai.djl.repository.Artifact
Returns the artifact arguments.
getArguments() - Method in class ai.djl.repository.Artifact
Returns the artifact arguments.
getArguments() - Method in class ai.djl.repository.zoo.Criteria
Returns the override configurations of the model loading arguments.
getArray() - Method in class ai.djl.nn.Parameter
Gets the values of this Parameter as an NDArray.
getArtifact(String, Function<InputStream, T>) - Method in class ai.djl.BaseModel
Attempts to load the artifact using the given function and cache it if the specified artifact is not already cached.
getArtifact(String) - Method in class ai.djl.BaseModel
Finds an artifact resource with a given name in the model.
getArtifact(String, Function<InputStream, T>) - Method in interface ai.djl.Model
Attempts to load the artifact using the given function and cache it if the specified artifact is not already cached.
getArtifact(String) - Method in interface ai.djl.Model
Finds an artifact resource with a given name in the model.
getArtifact() - Method in class ai.djl.repository.Artifact.Item
Returns the artifact associated with this item.
getArtifact(String, Function<InputStream, T>) - Method in class ai.djl.repository.zoo.ZooModel
Attempts to load the artifact using the given function and cache it if the specified artifact is not already cached.
getArtifact(String) - Method in class ai.djl.repository.zoo.ZooModel
Finds an artifact resource with a given name in the model.
getArtifactAsStream(String) - Method in class ai.djl.BaseModel
Finds an artifact resource with a given name in the model.
getArtifactAsStream(String) - Method in interface ai.djl.Model
Finds an artifact resource with a given name in the model.
getArtifactAsStream(String) - Method in class ai.djl.repository.zoo.ZooModel
Finds an artifact resource with a given name in the model.
getArtifactId() - Method in class ai.djl.repository.Metadata
Returns the artifactId.
getArtifactId() - Method in class ai.djl.repository.MRL
Returns the artifactId.
getArtifactId() - Method in class ai.djl.repository.zoo.BaseModelLoader
Returns the artifact ID of the ModelLoader.
getArtifactId() - Method in class ai.djl.repository.zoo.Criteria
Returns the artifactId of the ModelLoader to be searched.
getArtifactId() - Method in interface ai.djl.repository.zoo.ModelLoader
Returns the artifact ID of the ModelLoader.
getArtifactNames() - Method in class ai.djl.BaseModel
Returns the artifact names associated with the model.
getArtifactNames() - Method in interface ai.djl.Model
Returns the artifact names associated with the model.
getArtifactNames() - Method in class ai.djl.repository.zoo.ZooModel
Returns the artifact names associated with the model.
getArtifacts() - Method in class ai.djl.repository.Metadata
Returns all the artifacts in the metadata.
getAsBytes(String) - Method in class ai.djl.modality.Input
Returns the value as byte[] for the first key found in the Input.
getAsBytes(int) - Method in class ai.djl.modality.Input
Returns the value as byte[] at the specified position in the Input.
getAsBytes() - Method in interface ai.djl.ndarray.BytesSupplier
Returns the byte[] presentation of the object.
getAsBytes() - Method in class ai.djl.ndarray.NDList
Returns the byte[] presentation of the object.
getAsNDArray(NDManager, String) - Method in class ai.djl.modality.Input
Returns the value as NDArray for the first key found in the Input.
getAsNDArray(NDManager, int) - Method in class ai.djl.modality.Input
Returns the value as NDArray at the specified position in the Input.
getAsNDList(NDManager, String) - Method in class ai.djl.modality.Input
Returns the value as NDList for the first key found in the Input.
getAsNDList(NDManager, int) - Method in class ai.djl.modality.Input
Returns the value as NDList at the specified position in the Input.
getAsObject() - Method in interface ai.djl.ndarray.BytesSupplier
Returns the object that backs this BytesSupplier.
getAsString() - Method in class ai.djl.modality.Classifications
Returns the String presentation of the object.
getAsString(String) - Method in class ai.djl.modality.Input
Returns the value as byte[] for the first key found in the Input.
getAsString(int) - Method in class ai.djl.modality.Input
Returns the value as byte[] at the specified position in the Input.
getAsString() - Method in interface ai.djl.ndarray.BytesSupplier
Returns the String presentation of the object.
getAttachment(String) - Method in interface ai.djl.translate.TranslatorContext
Returns value of attached key-value pair to context.
getAttentionMask() - Method in class ai.djl.modality.nlp.bert.BertToken
Gets the mask to avoid performing attention on padding token indices.
getAxis() - Method in class ai.djl.ndarray.index.full.NDIndexFullPick
Returns the axis to pick.
getBaseUri() - Method in class ai.djl.repository.AbstractRepository
Returns the URI to the base of the repository.
getBaseUri() - Method in interface ai.djl.repository.Repository
Returns the URI to the base of the repository.
getBaseUri() - Method in class ai.djl.repository.SimpleRepository
Returns the URI to the base of the repository.
getBatch() - Method in interface ai.djl.modality.rl.env.RlEnv
Returns a batch of steps from the environment ReplayBuffer.
getBatch() - Method in class ai.djl.modality.rl.LruReplayBuffer
Returns a batch of steps from this buffer.
getBatch() - Method in interface ai.djl.modality.rl.ReplayBuffer
Returns a batch of steps from this buffer.
getBatch() - Method in class ai.djl.training.listener.TrainingListener.BatchData
Returns the original batch.
getBatchifier() - Method in class ai.djl.modality.cv.translator.BaseImageTranslator
Gets the Batchifier.
getBatchifier() - Method in class ai.djl.modality.cv.translator.ImageServingTranslator
Gets the Batchifier.
getBatchifier() - Method in class ai.djl.modality.cv.translator.wrapper.FileTranslator
Gets the Batchifier.
getBatchifier() - Method in class ai.djl.modality.cv.translator.wrapper.InputStreamTranslator
Gets the Batchifier.
getBatchifier() - Method in class ai.djl.modality.cv.translator.wrapper.UrlTranslator
Gets the Batchifier.
getBatchifier() - Method in class ai.djl.modality.nlp.translator.QaServingTranslator
Gets the Batchifier.
getBatchifier() - Method in class ai.djl.modality.nlp.translator.QATranslator
Gets the Batchifier.
getBatchifier() - Method in class ai.djl.modality.nlp.translator.SimpleText2TextTranslator
Gets the Batchifier.
getBatchifier() - Method in class ai.djl.modality.nlp.translator.TextClassificationServingTranslator
Gets the Batchifier.
getBatchifier() - Method in interface ai.djl.translate.NoBatchifyTranslator
 
getBatchifier() - Method in class ai.djl.translate.NoopTranslator
Gets the Batchifier.
getBatchifier() - Method in interface ai.djl.translate.Translator
Gets the Batchifier.
getBatchSize() - Method in class ai.djl.training.dataset.BatchSampler
Returns the batch size of the Sampler.
getBatchSize() - Method in interface ai.djl.training.dataset.Sampler
Returns the batch size of the Sampler.
getBertMaskedLanguageModelLoss() - Method in class ai.djl.nn.transformer.BertPretrainingLoss
gets BertMaskedLanguageModelLoss.
getBertNextSentenceLoss() - Method in class ai.djl.nn.transformer.BertPretrainingLoss
gets BertNextSentenceLoss.
getBest() - Method in class ai.djl.training.hyperparameter.optimizer.BaseHpOptimizer
Returns the best hyperparameters and loss.
getBest() - Method in interface ai.djl.training.hyperparameter.optimizer.HpOptimizer
Returns the best hyperparameters and loss.
getBlock() - Method in class ai.djl.BaseModel
Gets the block from the Model.
getBlock() - Method in interface ai.djl.Model
Gets the block from the Model.
getBlock() - Method in class ai.djl.repository.zoo.Criteria
Returns the optional Block to be used for ZooModel.
getBlock() - Method in class ai.djl.repository.zoo.ZooModel
Gets the block from the Model.
getBlock() - Method in interface ai.djl.translate.TranslatorContext
Returns the block from the TranslatorContext.
getBoolean(long...) - Method in interface ai.djl.ndarray.NDArray
Returns a boolean element from this NDArray.
getBoundingBox() - Method in class ai.djl.modality.cv.output.DetectedObjects.DetectedObject
Returns the BoundingBox of the detected object.
getBounds() - Method in interface ai.djl.modality.cv.output.BoundingBox
Returns the bounding Rectangle of this BoundingBox.
getBounds() - Method in class ai.djl.modality.cv.output.Rectangle
Returns the bounding Rectangle of this BoundingBox.
getByte(long...) - Method in interface ai.djl.ndarray.NDArray
Returns an byte element from this NDArray.
getCacheDirectory() - Method in class ai.djl.repository.AbstractRepository
Returns the cache directory for the repository.
getCacheDirectory() - Method in interface ai.djl.repository.Repository
Returns the cache directory for the repository.
getCacheDirectory() - Method in class ai.djl.repository.SimpleRepository
Returns the cache directory for the repository.
getCheckpoint() - Method in class ai.djl.training.listener.SaveModelTrainingListener
Returns the checkpoint frequency (or -1 for no checkpointing) in SaveModelTrainingListener.
getChildren() - Method in class ai.djl.nn.AbstractBlock
Returns a list of all the children of the block.
getChildren() - Method in interface ai.djl.nn.Block
Returns a list of all the children of the block.
getClassName() - Method in class ai.djl.modality.Classifications.Classification
Returns the class name.
getCode() - Method in class ai.djl.modality.Output
Returns the status code of the output.
getComponents() - Method in class ai.djl.training.loss.AbstractCompositeLoss
Returns the component losses that make up the composite loss.
getConfidence() - Method in class ai.djl.modality.cv.output.Joints.Joint
Returns the confidence probability for the joint.
getContent() - Method in class ai.djl.modality.Input
Returns the content of the input.
getData() - Method in class ai.djl.modality.Input
Returns the default data item.
getData() - Method in class ai.djl.training.dataset.Batch
Gets the data of this Batch.
getData(NDManager) - Method in interface ai.djl.training.dataset.Dataset
Fetches an iterator that can iterate through the Dataset.
getData(NDManager, ExecutorService) - Method in interface ai.djl.training.dataset.Dataset
Fetches an iterator that can iterate through the Dataset with multiple threads.
getData(NDManager) - Method in class ai.djl.training.dataset.RandomAccessDataset
Fetches an iterator that can iterate through the Dataset.
getData(NDManager, ExecutorService) - Method in class ai.djl.training.dataset.RandomAccessDataset
Fetches an iterator that can iterate through the Dataset with multiple threads.
getData(NDManager, Sampler) - Method in class ai.djl.training.dataset.RandomAccessDataset
Fetches an iterator that can iterate through the Dataset with a custom sampler.
getData(NDManager, Sampler, ExecutorService) - Method in class ai.djl.training.dataset.RandomAccessDataset
Fetches an iterator that can iterate through the Dataset with a custom sampler multi-threaded.
getData() - Method in class ai.djl.training.dataset.Record
Gets the data of this Record.
getDataAsNDList(NDManager) - Method in class ai.djl.modality.Input
Returns the default data as NDList.
getDataset(Dataset.Usage) - Method in class ai.djl.training.hyperparameter.EasyHpo
Returns the dataset to train with.
getDataType() - Method in class ai.djl.BaseModel
Returns the standard data type used within the model.
getDataType() - Method in interface ai.djl.Model
Returns the standard data type used within the model.
getDataType() - Method in interface ai.djl.ndarray.NDArray
Returns the DataType of this NDArray.
getDataType() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the DataType of this NDArray.
getDataType() - Method in class ai.djl.ndarray.types.DataDesc
Returns the DataType of the NDArray.
getDataType() - Method in class ai.djl.repository.zoo.ZooModel
Returns the standard data type used within the model.
getDefaultArtifact() - Method in class ai.djl.repository.MRL
Returns the default artifact.
getDefaultEngineName() - Static method in class ai.djl.engine.Engine
Returns the default Engine name.
getDescription() - Method in class ai.djl.repository.Metadata
Returns the description.
getDevice() - Method in class ai.djl.ndarray.BaseNDManager
Returns the default Device of this NDManager.
getDevice() - Method in interface ai.djl.ndarray.NDArray
Returns the Device of this NDArray.
getDevice() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the Device of this NDArray.
getDevice() - Method in interface ai.djl.ndarray.NDManager
Returns the default Device of this NDManager.
getDevice() - Method in class ai.djl.repository.zoo.Criteria
Returns the Device of the model to be loaded on.
getDeviceId() - Method in class ai.djl.Device
Returns the deviceId of the Device.
getDevices() - Method in class ai.djl.engine.Engine
Returns an array of devices.
getDevices(int) - Method in class ai.djl.engine.Engine
Returns an array of devices given the maximum number of GPUs to use.
getDevices() - Method in class ai.djl.training.DefaultTrainingConfig
Gets the Device that are available for computation.
getDevices() - Method in class ai.djl.training.Trainer
Returns the devices used for training.
getDevices() - Method in interface ai.djl.training.TrainingConfig
Gets the Device that are available for computation.
getDeviceType() - Method in class ai.djl.Device
Returns the device type of the Device.
getDirectParameters() - Method in class ai.djl.nn.AbstractBlock
Returns a list of all the direct parameters of the block.
getDirectParameters() - Method in interface ai.djl.nn.Block
Returns a list of all the direct parameters of the block.
getDouble(long...) - Method in interface ai.djl.ndarray.NDArray
Returns a double element from this NDArray.
getEllipsisIndex() - Method in class ai.djl.ndarray.index.NDIndex
Returns the index of the ellipsis.
getEmbeddingSize() - Method in class ai.djl.nn.transformer.BertBlock
Returns the embedding size used for tokens.
getEmbeddingType() - Method in class ai.djl.nn.core.Embedding.BaseBuilder
Returns the embedded type.
getEncoded() - Method in class ai.djl.ndarray.types.Shape
Gets the byte array representation of this Shape for serialization.
getEngine(String) - Static method in class ai.djl.engine.Engine
Returns the Engine with the given name.
getEngine() - Method in interface ai.djl.engine.EngineProvider
Returns the instance of the Engine class EngineProvider should bind to.
getEngine() - Method in interface ai.djl.ndarray.NDManager
Returns the Engine associated with this manager.
getEngine() - Method in class ai.djl.repository.zoo.Criteria
Returns the engine name.
getEngineName() - Method in class ai.djl.engine.Engine
Returns the name of the Engine.
getEngineName() - Method in interface ai.djl.engine.EngineProvider
Returns the name of the Engine.
getEngineRank() - Method in interface ai.djl.engine.EngineProvider
Returns the rank of the Engine.
getEpoch() - Method in class ai.djl.training.TrainingResult
Returns the actual number of epoch.
getEvaluations() - Method in class ai.djl.training.TrainingResult
Returns the raw evaluation metrics.
getEvaluators() - Method in class ai.djl.training.DefaultTrainingConfig
Returns the list of Evaluators that should be computed during training.
getEvaluators() - Method in class ai.djl.training.Trainer
Gets all Evaluators.
getEvaluators() - Method in interface ai.djl.training.TrainingConfig
Returns the list of Evaluators that should be computed during training.
getExecutorService() - Method in class ai.djl.training.DefaultTrainingConfig
 
getExecutorService() - Method in class ai.djl.training.Trainer
Returns the ExecutorService.
getExecutorService() - Method in interface ai.djl.training.TrainingConfig
Gets the ExecutorService for parallelization.
getExpectedLayout() - Method in class ai.djl.nn.convolutional.Conv1d
Returns the expected layout of the input.
getExpectedLayout() - Method in class ai.djl.nn.convolutional.Conv1dTranspose
Returns the expected layout of the input.
getExpectedLayout() - Method in class ai.djl.nn.convolutional.Conv2d
Returns the expected layout of the input.
getExpectedLayout() - Method in class ai.djl.nn.convolutional.Conv2dTranspose
Returns the expected layout of the input.
getExpectedLayout() - Method in class ai.djl.nn.convolutional.Conv3d
Returns the expected layout of the input.
getExpectedLayout() - Method in class ai.djl.nn.convolutional.Convolution
Returns the expected layout of the input.
getExpectedLayout() - Method in class ai.djl.nn.convolutional.Deconvolution
Returns the expected layout of the input.
getExtension() - Method in class ai.djl.repository.Artifact.Item
Returns the type of file extension.
getFile(Artifact.Item, String) - Method in class ai.djl.repository.AbstractRepository
Returns the path to a file for the item.
getFile(Artifact.Item, String) - Method in interface ai.djl.repository.Repository
Returns the path to a file for the item.
getFileExtension(String) - Static method in class ai.djl.repository.FilenameUtils
Returns the file name extension of the file.
getFiles() - Method in class ai.djl.repository.Artifact
Returns all the file items in the artifact.
getFileType(String) - Static method in class ai.djl.repository.FilenameUtils
Returns the type of the file.
getFilters() - Method in class ai.djl.repository.zoo.Criteria
Returns the search filters that must match the properties of the model.
getFloat(long...) - Method in interface ai.djl.ndarray.NDArray
Returns a float element from this NDArray.
getFormat() - Method in enum ai.djl.ndarray.types.DataType
Returns the format of the data type.
getGpuCount() - Method in class ai.djl.engine.Engine
Returns the number of GPUs available in the system.
getGradient() - Method in interface ai.djl.ndarray.NDArray
Returns the gradient NDArray attached to this NDArray.
getGradient() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the gradient NDArray attached to this NDArray.
getGroupId() - Method in class ai.djl.repository.Metadata
Returns the groupId.
getGroupId() - Method in class ai.djl.repository.MRL
Returns the groupId.
getGroupId() - Method in class ai.djl.repository.zoo.Criteria
Returns the groupId of the ModelZoo to be searched.
getGroupId() - Method in class ai.djl.repository.zoo.DefaultModelZoo
Returns the global unique identifier of the ModelZoo.
getGroupId() - Method in class ai.djl.repository.zoo.ModelZoo
Returns the global unique identifier of the ModelZoo.
getHeight() - Method in interface ai.djl.modality.cv.Image
Gets the height of the image.
getHeight() - Method in class ai.djl.modality.cv.output.Rectangle
Returns the height of the Rectangle.
getHParam(String) - Method in class ai.djl.training.hyperparameter.param.HpSet
Returns the hyperparameter in the set with the given name.
getId() - Method in class ai.djl.nn.Parameter
Gets the ID of this Parameter.
getId() - Method in class ai.djl.repository.License
Returns the identifier of the license.
getImageHeight() - Method in class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder
Get resized image height.
getImageWidth() - Method in class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder
Get resized image width.
getIncrementalVersion() - Method in class ai.djl.repository.Version
Returns the incremental version (assuming major.minor.incremental...) of the version.
getIndex(String) - Method in class ai.djl.modality.nlp.DefaultVocabulary
Returns the index of the given token.
getIndex(String) - Method in interface ai.djl.modality.nlp.Vocabulary
Returns the index of the given token.
getIndex() - Method in class ai.djl.ndarray.index.dim.NDIndexBooleans
Returns the mask binary NDArray.
getIndex() - Method in class ai.djl.ndarray.index.dim.NDIndexFixed
Returns the dimension of the index.
getIndices() - Method in class ai.djl.ndarray.index.dim.NDIndexPick
Returns the indices to pick.
getIndices() - Method in class ai.djl.ndarray.index.full.NDIndexFullPick
Returns the indices to pick.
getIndices() - Method in class ai.djl.ndarray.index.NDIndex
Returns the indices.
getInitializer() - Method in enum ai.djl.nn.Parameter.Type
Gets the Initializer of this ParameterType.
getInitializers() - Method in class ai.djl.training.DefaultTrainingConfig
Gets a list of Initializer and Predicate to initialize the parameters of the model.
getInitializers() - Method in interface ai.djl.training.TrainingConfig
Gets a list of Initializer and Predicate to initialize the parameters of the model.
getInputClass() - Method in class ai.djl.repository.zoo.Criteria
Returns the input data type.
getInstance() - Static method in class ai.djl.engine.Engine
Returns the default Engine.
getInstance() - Static method in class ai.djl.modality.cv.ImageFactory
Gets new instance of Image factory from the provided factory implementation.
getInt(long...) - Method in interface ai.djl.ndarray.NDArray
Returns an int element from this NDArray.
getIoU(BoundingBox) - Method in interface ai.djl.modality.cv.output.BoundingBox
Gets the Intersection over Union (IoU) value between bounding boxes.
getIoU(BoundingBox) - Method in class ai.djl.modality.cv.output.Rectangle
Gets the Intersection over Union (IoU) value between bounding boxes.
getJoints() - Method in class ai.djl.modality.cv.output.Joints
Gets the joints for the image.
getKeyProjection() - Method in class ai.djl.nn.transformer.ScaledDotProductAttentionBlock
Pointwise Linear projection of the keys.
getLabels() - Method in class ai.djl.training.dataset.Batch
Gets the labels corresponding to the data of this Batch.
getLabels() - Method in class ai.djl.training.dataset.Record
Gets the labels that correspond to the data of this Record.
getLabels() - Method in class ai.djl.training.listener.TrainingListener.BatchData
Returns the labels for each device.
getLastUpdated() - Method in class ai.djl.repository.Metadata
Returns the last update date for the metadata.
getLatestEvaluations() - Method in class ai.djl.training.listener.EvaluatorTrainingListener
Returns the latest evaluations.
getLayout() - Method in class ai.djl.ndarray.types.Shape
Returns the layout type for each axis in this shape.
getLayoutType(int) - Method in class ai.djl.ndarray.types.Shape
Returns the layout type in the given dimension.
getLeadingOnes() - Method in class ai.djl.ndarray.types.Shape
Returns the number of leading ones in the array shape.
getLicenses() - Method in class ai.djl.repository.Metadata
Returns the License.
getLong(long...) - Method in interface ai.djl.ndarray.NDArray
Returns a long element from this NDArray.
getLoss(HpSet) - Method in class ai.djl.training.hyperparameter.optimizer.BaseHpOptimizer
Returns the recorded loss.
getLoss(HpSet) - Method in interface ai.djl.training.hyperparameter.optimizer.HpOptimizer
Returns the recorded loss.
getLoss() - Method in class ai.djl.training.Trainer
Gets the training Loss function of the trainer.
getLossFunction() - Method in class ai.djl.training.DefaultTrainingConfig
Gets the Loss function to compute the loss against.
getLossFunction() - Method in interface ai.djl.training.TrainingConfig
Gets the Loss function to compute the loss against.
getMajorVersion() - Method in class ai.djl.repository.Version
Returns the major version (assuming major.minor.incremental...) of the version.
getManager() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the NDManager that manages this.
getManager() - Method in class ai.djl.ndarray.NDList
Returns the NDManager that manages this.
getManager() - Method in interface ai.djl.ndarray.NDResource
Returns the NDManager that manages this.
getManager() - Method in class ai.djl.training.dataset.Batch
Gets the NDManager that is attached to this Batch.
getManager() - Method in class ai.djl.training.ParameterStore
Get the NDManager associated with ParameterStore.
getManager() - Method in class ai.djl.training.Trainer
Gets the NDManager from the model.
getMax() - Method in class ai.djl.ndarray.index.dim.NDIndexSlice
Returns the end of the range.
getMax() - Method in class ai.djl.ndarray.index.full.NDIndexFullSlice
Returns the slice max for each axis.
getMax() - Method in class ai.djl.training.util.MinMaxScaler
Returns the value of fittedMax.
getMessage() - Method in class ai.djl.modality.Output
Returns the status code of the output.
getMetadata() - Method in class ai.djl.repository.Artifact
Returns the metadata containing this artifact.
getMetadataVersion() - Method in class ai.djl.repository.Artifact
Returns the metadata format version.
getMetadataVersion() - Method in class ai.djl.repository.Metadata
Returns the metadata format version.
getMetric(String) - Method in class ai.djl.metric.Metrics
Returns all Metrics with the specified metric name.
getMetricName() - Method in class ai.djl.metric.Metric
Returns the name of the Metric.
getMetricNames() - Method in class ai.djl.metric.Metrics
Returns a set of String metric names.
getMetrics() - Method in class ai.djl.training.Trainer
Returns the Metrics param used for benchmarking.
getMetrics() - Method in interface ai.djl.translate.TranslatorContext
Returns the Metric tool to do benchmark.
getMin() - Method in class ai.djl.ndarray.index.dim.NDIndexSlice
Returns the start of the range.
getMin() - Method in class ai.djl.ndarray.index.full.NDIndexFullSlice
Returns the slice min for each axis.
getMin() - Method in class ai.djl.training.util.MinMaxScaler
Returns the value of fittedMin.
getMinorVersion() - Method in class ai.djl.repository.Version
Returns the minor version (assuming major.minor.incremental...) of the version.
getModel() - Method in class ai.djl.training.Trainer
Returns the model used to create this trainer.
getModel() - Method in interface ai.djl.translate.TranslatorContext
Returns the Model object to understand the input/output.
getModelLoader(String) - Method in class ai.djl.repository.zoo.ModelZoo
Returns the ModelLoader based on the model name.
getModelLoaders() - Method in class ai.djl.repository.zoo.DefaultModelZoo
Lists the available model families in the ModelZoo.
getModelLoaders() - Method in class ai.djl.repository.zoo.ModelZoo
Lists the available model families in the ModelZoo.
getModelName() - Method in class ai.djl.repository.zoo.Criteria
Returns the optional model name to be used for ZooModel.
getModelPath() - Method in class ai.djl.BaseModel
Returns the directory from where the model is loaded.
getModelPath() - Method in interface ai.djl.Model
Returns the directory from where the model is loaded.
getModelPath() - Method in class ai.djl.repository.zoo.ZooModel
Returns the directory from where the model is loaded.
getModelZoo() - Method in class ai.djl.repository.zoo.Criteria
Returns the ModelZoo to be searched.
getModelZoo() - Method in class ai.djl.repository.zoo.DefaultZooProvider
Returns the instance of the ModelZoo.
getModelZoo(String) - Static method in class ai.djl.repository.zoo.ModelZoo
Returns the ModelZoo with the groupId.
getModelZoo() - Method in interface ai.djl.repository.zoo.ZooProvider
Returns the instance of the ModelZoo.
getName() - Method in class ai.djl.BaseModel
Gets the model name.
getName() - Method in interface ai.djl.Model
Gets the model name.
getName() - Method in class ai.djl.ndarray.BaseNDManager
Gets the name of the NDManager.
getName() - Method in interface ai.djl.ndarray.NDArray
Returns the name of this NDArray.
getName() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the name of this NDArray.
getName() - Method in interface ai.djl.ndarray.NDManager
Gets the name of the NDManager.
getName() - Method in class ai.djl.ndarray.types.DataDesc
Returns the name of the NDArray.
getName() - Method in class ai.djl.nn.Parameter
Gets the name of this Parameter.
getName() - Method in class ai.djl.repository.AbstractRepository
Returns the repository name.
getName() - Method in class ai.djl.repository.Artifact
Returns the artifact name.
getName() - Method in class ai.djl.repository.Artifact.Item
Returns the item name.
getName() - Method in class ai.djl.repository.License
Returns the name of the license.
getName() - Method in class ai.djl.repository.Metadata
Returns the metadata-level name.
getName() - Method in interface ai.djl.repository.Repository
Returns the repository name.
getName() - Method in class ai.djl.repository.zoo.ZooModel
Gets the model name.
getName() - Method in interface ai.djl.repository.zoo.ZooProvider
Gets the name of the ModelZoo.
getName() - Method in class ai.djl.training.evaluator.Evaluator
Returns the name of this Evaluator.
getName() - Method in class ai.djl.training.hyperparameter.param.Hyperparameter
Returns the name of the hyperparameter.
getNamePart(String) - Static method in class ai.djl.repository.FilenameUtils
Returns the name of the file without file extension.
getNDArrayInternal() - Method in interface ai.djl.ndarray.NDArray
Returns an internal representative of Native NDArray.
getNDArrayInternal() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns an internal representative of Native NDArray.
getNDManager() - Method in class ai.djl.BaseModel
Gets the NDManager from the model.
getNDManager() - Method in interface ai.djl.Model
Gets the NDManager from the model.
getNDManager() - Method in class ai.djl.repository.zoo.ZooModel
Gets the NDManager from the model.
getNDManager() - Method in interface ai.djl.translate.TranslatorContext
Returns the NDManager to create NDArray.
getNewValue(int) - Method in class ai.djl.training.tracker.CosineTracker
Fetches the value after the given number of steps/updates.
getNewValue(int) - Method in class ai.djl.training.tracker.FactorTracker
Fetches the value after the given number of steps/updates.
getNewValue(int) - Method in class ai.djl.training.tracker.LinearTracker
Fetches the value after the given number of steps/updates.
getNewValue(int) - Method in class ai.djl.training.tracker.MultiFactorTracker
Fetches the value after the given number of steps/updates.
getNewValue(int) - Method in class ai.djl.training.tracker.PolynomialDecayTracker
Fetches the value after the given number of steps/updates.
getNewValue(int) - Method in interface ai.djl.training.tracker.Tracker
Fetches the value after the given number of steps/updates.
getNewValue(int) - Method in class ai.djl.training.tracker.WarmUpTracker
Fetches the value after the given number of steps/updates.
getNumberOfObjects() - Method in class ai.djl.modality.cv.output.DetectedObjects
Returns the number of objects found in an image.
getNumDirections() - Method in class ai.djl.nn.recurrent.RecurrentBlock
 
getNumEpochs() - Method in class ai.djl.training.listener.EpochTrainingListener
Returns the number of epochs.
getNumOfBytes() - Method in enum ai.djl.ndarray.types.DataType
Returns the number of bytes for each element.
getObservation() - Method in interface ai.djl.modality.rl.env.RlEnv
Returns the observation detailing the current state of the environment.
getOptimizer() - Method in class ai.djl.training.DefaultTrainingConfig
Gets the Optimizer to use during training.
getOptimizer() - Method in interface ai.djl.training.TrainingConfig
Gets the Optimizer to use during training.
getOptions(Map<String, String>) - Method in class ai.djl.repository.Artifact
Returns the artifact options.
getOptions() - Method in class ai.djl.repository.zoo.Criteria
Returns the model loading options.
getOutputClass() - Method in class ai.djl.repository.zoo.Criteria
Returns the output data type.
getOutputShapes(Shape[]) - Method in class ai.djl.modality.nlp.Decoder
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.modality.nlp.embedding.TrainableTextEmbedding
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.modality.nlp.Encoder
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.modality.nlp.EncoderDecoder
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.AbstractSymbolBlock
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in interface ai.djl.nn.Block
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.convolutional.Convolution
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.convolutional.Deconvolution
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.core.ConstantEmbedding
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.core.Embedding
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.core.Linear
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.core.Prelu
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.LambdaBlock
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.norm.BatchNorm
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.norm.Dropout
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.norm.LayerNorm
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.ParallelBlock
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.recurrent.RecurrentBlock
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.SequentialBlock
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.transformer.BertBlock
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.transformer.BertMaskedLanguageModelBlock
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.transformer.BertNextSentenceBlock
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.transformer.BertPretrainingBlock
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.transformer.IdEmbedding
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.transformer.ScaledDotProductAttentionBlock
Returns the expected output shapes of the block for the specified input shapes.
getOutputShapes(Shape[]) - Method in class ai.djl.nn.transformer.TransformerEncoderBlock
Returns the expected output shapes of the block for the specified input shapes.
getOverrideModelName() - Method in class ai.djl.training.listener.SaveModelTrainingListener
Returns the override model name to save checkpoints with.
getParagraph() - Method in class ai.djl.modality.nlp.qa.QAInput
Gets the resource document that contains the answer.
getParameters() - Method in class ai.djl.nn.AbstractBlock
Returns a list of all the parameters of the block, including the parameters of its children fetched recursively.
getParameters() - Method in interface ai.djl.nn.Block
Returns a list of all the parameters of the block, including the parameters of its children fetched recursively.
getParentManager() - Method in class ai.djl.ndarray.BaseNDManager
Returns the parent NDManager.
getParentManager() - Method in interface ai.djl.ndarray.NDManager
Returns the parent NDManager.
getParsedVersion() - Method in class ai.djl.repository.Artifact
Returns the artifact version as a Version.
getPath() - Method in class ai.djl.Application
Returns the repository path of the application.
getPath() - Method in interface ai.djl.modality.cv.output.BoundingBox
Returns an iterator object that iterates along the BoundingBox boundary and provides access to the geometry of the BoundingBox outline.
getPath() - Method in class ai.djl.modality.cv.output.Landmark
Returns an iterator object that iterates along the BoundingBox boundary and provides access to the geometry of the BoundingBox outline.
getPath() - Method in class ai.djl.modality.cv.output.Rectangle
Returns an iterator object that iterates along the BoundingBox boundary and provides access to the geometry of the BoundingBox outline.
getPoint() - Method in interface ai.djl.modality.cv.output.BoundingBox
Returns the top left point of the bounding box.
getPoint() - Method in class ai.djl.modality.cv.output.Rectangle
Returns the top left point of the bounding box.
getPostActionSpace() - Method in interface ai.djl.modality.rl.env.RlEnv.Step
Returns the available actions after the step.
getPostObservation() - Method in interface ai.djl.modality.rl.env.RlEnv.Step
Returns the observation detailing the state after the action.
getPredictions() - Method in class ai.djl.training.listener.TrainingListener.BatchData
Returns the predictions for each device.
getPredictorManager() - Method in interface ai.djl.translate.TranslatorContext
Returns the Predictor's NDManager.
getPreObservation() - Method in interface ai.djl.modality.rl.env.RlEnv.Step
Returns the observation detailing the state before the action.
getPreprocessors(boolean) - Static method in class ai.djl.modality.nlp.bert.BertFullTokenizer
Get a list of TextProcessors to process input text for Bert models.
getProbability() - Method in class ai.djl.modality.Classifications.Classification
Returns the probability.
getProbDist() - Method in class ai.djl.modality.cv.output.Mask
Returns the probability for each pixel.
getProgress() - Method in class ai.djl.repository.zoo.Criteria
Returns the optional Progress for the model loading.
getProgress() - Method in class ai.djl.training.dataset.Batch
Returns the progress of the batch if it is part of some kind of iteration like a dataset iteration.
getProgressTotal() - Method in class ai.djl.training.dataset.Batch
Returns the total or end value for the progress of the batch if it is part of some kind of iteration like a dataset iteration.
getProperties() - Method in class ai.djl.modality.Input
Returns the properties of the input.
getProperties() - Method in class ai.djl.repository.Artifact
Returns the artifact properties.
getProperty(String) - Method in class ai.djl.BaseModel
Gets the property of the model based on property name.
getProperty(String, String) - Method in class ai.djl.modality.Input
Returns the value to which the specified key is mapped.
getProperty(String) - Method in interface ai.djl.Model
Gets the property of the model based on property name.
getProperty(String) - Method in class ai.djl.repository.zoo.ZooModel
Gets the property of the model based on property name.
getQueryProjection() - Method in class ai.djl.nn.transformer.ScaledDotProductAttentionBlock
Pointwise Linear projection of the queries.
getQuestion() - Method in class ai.djl.modality.nlp.qa.QAInput
Gets the question for the model.
getRank() - Method in class ai.djl.engine.Engine
Return the rank of the Engine.
getRank() - Method in class ai.djl.ndarray.index.dim.NDIndexAll
Returns the number of dimensions occupied by this index element.
getRank() - Method in class ai.djl.ndarray.index.dim.NDIndexBooleans
Returns the number of dimensions occupied by this index element.
getRank() - Method in interface ai.djl.ndarray.index.dim.NDIndexElement
Returns the number of dimensions occupied by this index element.
getRank() - Method in class ai.djl.ndarray.index.dim.NDIndexFixed
Returns the number of dimensions occupied by this index element.
getRank() - Method in class ai.djl.ndarray.index.dim.NDIndexPick
 
getRank() - Method in class ai.djl.ndarray.index.dim.NDIndexSlice
Returns the number of dimensions occupied by this index element.
getRank() - Method in class ai.djl.ndarray.index.NDIndex
Returns the number of dimensions specified in the Index.
getRecommendedVersion() - Method in class ai.djl.repository.VersionRange
Returns the recommended version in the range.
getRepository() - Method in class ai.djl.repository.MRL
Returns the repository.
getRepositoryUri() - Method in class ai.djl.repository.Metadata
Returns the URI to the repository storing the metadata.
getResourceDirectory(Artifact) - Method in interface ai.djl.repository.Repository
Returns the resource directory for the an artifact.
getResourceDirectory(Artifact) - Method in class ai.djl.repository.SimpleRepository
Returns the resource directory for the an artifact.
getResources() - Method in class ai.djl.repository.JarRepository
Returns a list of MRLs in the repository.
getResources() - Method in class ai.djl.repository.LocalRepository
Returns a list of MRLs in the repository.
getResources() - Method in class ai.djl.repository.RemoteRepository
Returns a list of MRLs in the repository.
getResources() - Method in interface ai.djl.repository.Repository
Returns a list of MRLs in the repository.
getResources() - Method in class ai.djl.repository.SimpleRepository
Returns a list of MRLs in the repository.
getResources() - Method in class ai.djl.repository.SimpleUrlRepository
Returns a list of MRLs in the repository.
getResourceType() - Method in class ai.djl.repository.Metadata
Returns the resource type.
getResourceUri() - Method in class ai.djl.repository.Artifact
Returns the location of the resource directory.
getRestrictions() - Method in class ai.djl.repository.VersionRange
Returns the restrictions that compose the range.
getResultProjection() - Method in class ai.djl.nn.transformer.ScaledDotProductAttentionBlock
Pointwise Linear projection of the results.
getReward() - Method in interface ai.djl.modality.rl.env.RlEnv.Step
Returns the reward given for the action.
getSampler() - Method in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
Gets the Sampler for the dataset.
getScalar(long...) - Method in interface ai.djl.ndarray.NDArray
Returns a scalar NDArray corresponding to a single element.
getSeed() - Method in class ai.djl.engine.Engine
Returns the random seed in DJL Engine.
getSha1Hash() - Method in class ai.djl.repository.Artifact.Item
Returns the hash of the item.
getShape() - Method in class ai.djl.ndarray.index.full.NDIndexFullSlice
Returns the slice shape without squeezing.
getShape() - Method in interface ai.djl.ndarray.NDArray
Returns the Shape of this NDArray.
getShape() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the Shape of this NDArray.
getShape() - Method in class ai.djl.ndarray.types.DataDesc
Returns the Shape of the NDArray.
getShape() - Method in class ai.djl.ndarray.types.Shape
Returns the dimensions of the Shape.
getShapeFromEmptyNDArrayForReductionOp(Shape, int) - Static method in class ai.djl.ndarray.NDUtils
Get Shape of the empty NDArray after applying reduction operations.
getShapes() - Method in class ai.djl.ndarray.NDList
Gets all of shapes in the NDList.
getSize() - Method in class ai.djl.repository.Artifact.Item
Returns the file size.
getSize() - Method in class ai.djl.training.dataset.Batch
Returns the batchSize.
getSparseFormat() - Method in interface ai.djl.ndarray.NDArray
Returns the SparseFormat of this NDArray.
getSparseFormat() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the SparseFormat of this NDArray.
getSqueezedShape() - Method in class ai.djl.ndarray.index.full.NDIndexFullSlice
Returns the slice shape with squeezing.
getStates(NDList) - Method in class ai.djl.modality.nlp.Encoder
Gets the state of the encoder from the given encoder output.
getStep() - Method in class ai.djl.ndarray.index.dim.NDIndexSlice
Returns the step between each slice.
getStep() - Method in class ai.djl.ndarray.index.full.NDIndexFullSlice
Returns the slice step for each axis.
getStringLayout() - Method in class ai.djl.nn.convolutional.Conv1d
Returns the string representing the layout of the input.
getStringLayout() - Method in class ai.djl.nn.convolutional.Conv1dTranspose
Returns the string representing the layout of the input.
getStringLayout() - Method in class ai.djl.nn.convolutional.Conv2d
Returns the string representing the layout of the input.
getStringLayout() - Method in class ai.djl.nn.convolutional.Conv2dTranspose
Returns the string representing the layout of the input.
getStringLayout() - Method in class ai.djl.nn.convolutional.Conv3d
Returns the string representing the layout of the input.
getStringLayout() - Method in class ai.djl.nn.convolutional.Convolution
Returns the string representing the layout of the input.
getStringLayout() - Method in class ai.djl.nn.convolutional.Deconvolution
Returns the string representing the layout of the input.
getSubImage(int, int, int, int) - Method in interface ai.djl.modality.cv.Image
Gets the subimage defined by a specified rectangular region.
getSupportedEngines() - Method in class ai.djl.repository.zoo.DefaultModelZoo
Returns all supported engine names.
getSupportedEngines() - Method in class ai.djl.repository.zoo.ModelZoo
Returns all supported engine names.
getSupportedScheme() - Method in interface ai.djl.repository.RepositoryFactory
Returns a set of URI scheme that the RepositoryFactory supports.
getSupportedTypes() - Method in class ai.djl.modality.cv.translator.BigGANTranslatorFactory
Returns supported input/output classes.
getSupportedTypes() - Method in class ai.djl.modality.cv.translator.ImageClassificationTranslatorFactory
Returns supported input/output classes.
getSupportedTypes() - Method in class ai.djl.modality.cv.translator.ObjectDetectionTranslatorFactory
Returns supported input/output classes.
getSupportedTypes() - Method in class ai.djl.modality.cv.translator.SimplePoseTranslatorFactory
Returns supported input/output classes.
getSupportedTypes() - Method in class ai.djl.modality.cv.translator.StyleTransferTranslatorFactory
Returns supported input/output classes.
getSupportedTypes() - Method in class ai.djl.translate.DefaultTranslatorFactory
Returns supported input/output classes.
getSupportedTypes() - Method in class ai.djl.translate.ServingTranslatorFactory
Returns supported input/output classes.
getSupportedTypes() - Method in interface ai.djl.translate.TranslatorFactory
Returns supported input/output classes.
getTimestamp() - Method in class ai.djl.metric.Metric
Returns the timestamp of the Metric.
getToken(long) - Method in class ai.djl.modality.nlp.DefaultVocabulary
Returns the token corresponding to the given index.
getToken(long) - Method in interface ai.djl.modality.nlp.Vocabulary
Returns the token corresponding to the given index.
getTokenDictionarySize() - Method in class ai.djl.nn.transformer.BertBlock
Returns the size of the token dictionary.
getTokenEmbedding() - Method in class ai.djl.nn.transformer.BertBlock
Returns the token embedding used by this Bert model.
getTokens() - Method in class ai.djl.modality.nlp.bert.BertToken
Gets the indices of input sequence tokens in the vocabulary.
getTokenTypes() - Method in class ai.djl.modality.nlp.bert.BertToken
Gets segment token indices to indicate first and second portions of the inputs.
getToSqueeze() - Method in class ai.djl.ndarray.index.full.NDIndexFullSlice
Returns the squeeze array of axis.
getTrailingOnes() - Method in class ai.djl.ndarray.types.Shape
Returns the number of trailing ones in the array shape.
getTrainEvaluation(String) - Method in class ai.djl.training.TrainingResult
Returns the evaluation to which the specified key is mapped.
getTrainingListeners() - Method in class ai.djl.training.DefaultTrainingConfig
Returns the list of TrainingListeners that should be used during training.
getTrainingListeners() - Method in interface ai.djl.training.TrainingConfig
Returns the list of TrainingListeners that should be used during training.
getTrainingResult() - Method in class ai.djl.training.Trainer
Returns the TrainingResult.
getTrainLoss() - Method in class ai.djl.training.TrainingResult
Returns the train loss.
getTranslator() - Method in class ai.djl.repository.zoo.ZooModel
Returns the default translator.
getTranslatorFactory(Criteria<?, ?>, Map<String, Object>) - Method in class ai.djl.repository.zoo.BaseModelLoader
 
getTranslatorFactory() - Method in class ai.djl.repository.zoo.Criteria
Returns the optional TranslatorFactory to be used for ZooModel.
getType() - Method in enum ai.djl.ndarray.types.SparseFormat
Returns the SparseFormat name.
getType() - Method in class ai.djl.nn.Parameter
Gets the type of this Parameter.
getType() - Method in class ai.djl.repository.Artifact.Item
Sets the type of the item.
getTypeDictionarySize() - Method in class ai.djl.nn.transformer.BertBlock
Returns the size of the type dictionary.
getUid() - Method in interface ai.djl.ndarray.NDArray
Returns unique identifier of this NDArray.
getUid() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns unique identifier of this NDArray.
getUint8(long...) - Method in interface ai.djl.ndarray.NDArray
Returns an integer element from this NDArray that represent unsigned integer with 8 bits.
getUnit() - Method in class ai.djl.metric.Metric
Returns the unit of the Metric.
getUnknownValueCount() - Method in class ai.djl.ndarray.types.Shape
Return the count of unknown value in this Shape.
getUri() - Method in class ai.djl.repository.Artifact.Item
Returns the URI of the item.
getUrl() - Method in class ai.djl.repository.License
Returns the url of the license.
getValidateEvaluation(String) - Method in class ai.djl.training.TrainingResult
Returns the evaluation to which the specified key is mapped.
getValidateLoss() - Method in class ai.djl.training.TrainingResult
Returns the validate loss.
getValidLength() - Method in class ai.djl.modality.nlp.bert.BertToken
Gets the length of the original sentence which has question and paragraph.
getValue() - Method in class ai.djl.metric.Metric
Returns the value of the Metric.
getValue() - Method in enum ai.djl.ndarray.types.LayoutType
Returns the character representation of the layout type.
getValue() - Method in enum ai.djl.ndarray.types.SparseFormat
Returns the integer value of this SparseFormat.
getValue(ParameterStore, Device, boolean) - Method in class ai.djl.nn.transformer.IdEmbedding
Quick hack for bert model to acces embedding table, replace by a proper function to calculate raw logits from embeddings.
getValue(Parameter, Device, boolean) - Method in class ai.djl.training.ParameterStore
Returns the value of a mirrored parameter on a device.
getValueProjection() - Method in class ai.djl.nn.transformer.ScaledDotProductAttentionBlock
Pointwise Linear projection of the values.
getVersion() - Method in class ai.djl.engine.Engine
Returns the version of the deep learning engine.
getVersion() - Method in class ai.djl.repository.Artifact
Returns the artifact version.
getVersion() - Method in class ai.djl.repository.MRL
Returns the version.
getVocabulary() - Method in class ai.djl.modality.nlp.bert.BertFullTokenizer
Returns the Vocabulary used for tokenization.
getWebsite() - Method in class ai.djl.repository.Metadata
Returns the website.
getWeightDecay() - Method in class ai.djl.training.optimizer.Optimizer
Gets the value of weight decay.
getWidth() - Method in interface ai.djl.modality.cv.Image
Gets the width of the image.
getWidth() - Method in class ai.djl.modality.cv.output.Rectangle
Returns the width of the Rectangle.
getWrappedImage() - Method in interface ai.djl.modality.cv.Image
Gets the wrapped image.
getWrappedModel() - Method in class ai.djl.repository.zoo.ZooModel
Returns the wrapped model.
getX() - Method in class ai.djl.modality.cv.output.Point
Returns the X coordinate of this Point in double precision.
getX() - Method in class ai.djl.modality.cv.output.Rectangle
Returns the left x-coordinate of the Rectangle.
getY() - Method in class ai.djl.modality.cv.output.Point
Returns the Y coordinate of this Point in double precision.
getY() - Method in class ai.djl.modality.cv.output.Rectangle
Returns the top y-coordinate of the Rectangle.
globalAvgPool1d(NDArray) - Static method in class ai.djl.nn.pooling.Pool
Performs 1-D Global Avg Pooling on the input.
globalAvgPool1dBlock() - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the globalAvgPool1d pooling function.
globalAvgPool2d(NDArray) - Static method in class ai.djl.nn.pooling.Pool
Performs 2-D Global Avg Pooling on the input.
globalAvgPool2dBlock() - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the globalAvgPool2d pooling function.
globalAvgPool3d(NDArray) - Static method in class ai.djl.nn.pooling.Pool
Performs 3-D Global Avg Pooling on the input.
globalAvgPool3dBlock() - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the globalAvgPool3d pooling function.
globalLpPool1d(NDArray, float) - Static method in class ai.djl.nn.pooling.Pool
Performs 1-D Global LP Pooling on the input.
globalLpPool1dBlock(float) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the globalLpPool1d pooling function.
globalLpPool2d(NDArray, float) - Static method in class ai.djl.nn.pooling.Pool
Performs 2-D Global LP Pooling on the input.
globalLpPool2dBlock(float) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the globalLpPool2d pooling function.
globalLpPool3d(NDArray, float) - Static method in class ai.djl.nn.pooling.Pool
Performs 3-D Global LP Pooling on the input.
globalLpPool3dBlock(float) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the globalLpPool3d pooling function.
globalMaxPool1d(NDArray) - Static method in class ai.djl.nn.pooling.Pool
Performs 1-D Global Max Pooling on the input.
globalMaxPool1dBlock() - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the globalmaxPool1dBlock pooling function.
globalMaxPool2d(NDArray) - Static method in class ai.djl.nn.pooling.Pool
Performs 2-D Global Max Pooling on the input.
globalMaxPool2dBlock() - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the globalmaxPool2dBlock pooling function.
globalMaxPool3d(NDArray) - Static method in class ai.djl.nn.pooling.Pool
Performs 3-D Global Max Pooling on the input.
globalMaxPool3dBlock() - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the globalmaxPool3dBlock pooling function.
gpu() - Static method in class ai.djl.Device
Returns the default GPU Device.
gpu(int) - Static method in class ai.djl.Device
Returns a new instance of GPU Device with the specified deviceId.
GPU - Static variable in interface ai.djl.Device.Type
 
GradientCollector - Interface in ai.djl.training
An interface that provides a mechanism to collect gradients during training.
GROUP_ID - Static variable in class ai.djl.repository.zoo.DefaultModelZoo
 
groupId - Variable in class ai.djl.repository.Metadata
 
groups - Variable in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
 
groups - Variable in class ai.djl.nn.convolutional.Convolution
 
groups - Variable in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
 
groups - Variable in class ai.djl.nn.convolutional.Deconvolution
 
GRU - Class in ai.djl.nn.recurrent
GRU is an abstract implementation of recurrent neural networks which applies GRU (Gated Recurrent Unit) recurrent layer to input.
GRU.Builder - Class in ai.djl.nn.recurrent
The Builder to construct a GRU type of Block.
gt(Number) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Greater" comparison.
gt(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Greater Than" comparison.
gt(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Greater" comparison.
gt(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Greater Than" comparison.
gt(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Greater Than" comparison.
gt(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Greater Than" comparison.
gt(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Greater Than" comparison.
gte(Number) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Greater or equals" comparison.
gte(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Greater or equals" comparison.
gte(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Greater or equals" comparison.
gte(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Greater or equals" comparison.
gte(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Greater or equals" comparison.
gte(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Greater or equals" comparison.
gte(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Greater or equals" comparison.

H

hasBiases - Variable in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
 
hasBiases - Variable in class ai.djl.nn.recurrent.RecurrentBlock
 
hasCapability(String) - Method in class ai.djl.engine.Engine
Returns whether the engine has the specified capability.
hasEngine(String) - Static method in class ai.djl.engine.Engine
Returns if the specified engine is available.
hasGradient() - Method in interface ai.djl.ndarray.NDArray
Returns true if the gradient calculation is required for this NDArray.
hasGradient() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns true if the gradient calculation is required for this NDArray.
hashCode() - Method in class ai.djl.Application
hashCode() - Method in class ai.djl.Device
hashCode() - Method in class ai.djl.ndarray.types.Shape
hashCode() - Method in class ai.djl.repository.Version
hasItem(String) - Method in class ai.djl.modality.nlp.embedding.TrainableWordEmbedding
Returns whether an item is in the embedding.
hasItem(T) - Method in interface ai.djl.nn.core.AbstractEmbedding
Returns whether an item is in the embedding.
hasItem(Object) - Method in class ai.djl.nn.core.ConstantEmbedding
Returns whether an item is in the embedding.
hasItem(T) - Method in class ai.djl.nn.core.Embedding.DefaultEmbedding
Returns whether an item is in the embedding.
hasItem(T) - Method in class ai.djl.nn.core.Embedding.DefaultItem
Returns whether an item is in the embedding.
hasMetric(String) - Method in class ai.djl.metric.Metrics
Returns true if the metrics object has a metric with the given name.
hasModelZoo(String) - Static method in class ai.djl.repository.zoo.ModelZoo
Returns whether a model zoo with the group id is available.
hasNext() - Method in class ai.djl.training.dataset.DataIterable
hasProperties(Map<String, String>) - Method in class ai.djl.repository.Artifact
Returns true if every filter matches the corresponding property.
hasZeroDimension() - Method in class ai.djl.ndarray.types.Shape
Returns true if the NDArray contains zero dimensions.
head() - Method in class ai.djl.ndarray.NDList
Returns the head index of the NDList.
head() - Method in class ai.djl.ndarray.types.Shape
Returns the head index of the shape.
height - Variable in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
 
HingeLoss - Class in ai.djl.training.loss
HingeLoss is a type of Loss.
HingeLoss() - Constructor for class ai.djl.training.loss.HingeLoss
Calculates Hinge loss.
HingeLoss(String) - Constructor for class ai.djl.training.loss.HingeLoss
Calculates Hinge loss.
HingeLoss(String, int, float) - Constructor for class ai.djl.training.loss.HingeLoss
Calculates Hinge loss.
hingeLoss() - Static method in class ai.djl.training.loss.Loss
Returns a new instance of HingeLoss with default arguments.
hingeLoss(String) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of HingeLoss with default arguments.
hingeLoss(String, int, float) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of HingeLoss with the given arguments.
HpBool - Class in ai.djl.training.hyperparameter.param
A Hyperparameter for a boolean option.
HpBool(String) - Constructor for class ai.djl.training.hyperparameter.param.HpBool
Constructs a HpBool.
HpCategorical<T> - Class in ai.djl.training.hyperparameter.param
A Hyperparameter which is one of a fixed number of options (similar to an enum).
HpCategorical(String, List<T>) - Constructor for class ai.djl.training.hyperparameter.param.HpCategorical
Constructs a HpCategorical.
HpFloat - Class in ai.djl.training.hyperparameter.param
A Hyperparameter for a float.
HpFloat(String, float, float, boolean) - Constructor for class ai.djl.training.hyperparameter.param.HpFloat
Constructs a HpFloat.
HpInt - Class in ai.djl.training.hyperparameter.param
A Hyperparameter for an integer.
HpInt(String, int, int) - Constructor for class ai.djl.training.hyperparameter.param.HpInt
Constructs a HpInt.
HpOptimizer - Interface in ai.djl.training.hyperparameter.optimizer
An optimizer for Hyperparameters.
HpORandom - Class in ai.djl.training.hyperparameter.optimizer
A simple HpOptimizer that tries random hyperparameter choices within the range.
HpORandom(HpSet) - Constructor for class ai.djl.training.hyperparameter.optimizer.HpORandom
Constructs a new HpORandom.
HpSet - Class in ai.djl.training.hyperparameter.param
A nestable set of Hyperparameters.
HpSet(String, List<Hyperparameter<?>>) - Constructor for class ai.djl.training.hyperparameter.param.HpSet
Cosntructs a new HpSet.
HpSet(String) - Constructor for class ai.djl.training.hyperparameter.param.HpSet
Cosntructs a new empty HpSet.
HpVal<T> - Class in ai.djl.training.hyperparameter.param
A Hyperparameter with a known value instead of a range of possible values.
HpVal(String, T) - Constructor for class ai.djl.training.hyperparameter.param.HpVal
Cosntructs a new HpVal.
Hyperparameter<T> - Class in ai.djl.training.hyperparameter.param
A class representing an input to the network that can't be differentiated.
Hyperparameter(String) - Constructor for class ai.djl.training.hyperparameter.param.Hyperparameter
Constructs a hyperparameter with the given name.
hyperParams - Variable in class ai.djl.training.hyperparameter.optimizer.BaseHpOptimizer
 
HyphenNormalizer - Class in ai.djl.modality.nlp.preprocess
Unicode normalization does not take care of "exotic" hyphens that we normally do not want in NLP input.
HyphenNormalizer() - Constructor for class ai.djl.modality.nlp.preprocess.HyphenNormalizer
 

I

IdEmbedding - Class in ai.djl.nn.transformer
An Embedding from integer ids to float vectors.
IdEmbedding.Builder - Class in ai.djl.nn.transformer
The Builder to construct an IdEmbedding type of Block.
identityBlock() - Static method in class ai.djl.nn.Blocks
Creates a LambdaBlock that performs the identity function.
Image - Interface in ai.djl.modality.cv
Image is a container of an image in DJL.
Image.Flag - Enum in ai.djl.modality.cv
Flag indicates the color channel options for images.
Image.Interpolation - Enum in ai.djl.modality.cv
Interpolation indicates the Interpolation options for resizinig an image.
IMAGE_CLASSIFICATION - Static variable in interface ai.djl.Application.CV
An application where images are assigned a single class name.
IMAGE_ENHANCEMENT - Static variable in interface ai.djl.Application.CV
An application that accepts an image and returns enhanced images.
IMAGE_GENERATION - Static variable in interface ai.djl.Application.CV
An application that accepts a seed and returns generated images.
ImageClassificationTranslator - Class in ai.djl.modality.cv.translator
A generic Translator for Image Classification tasks.
ImageClassificationTranslator(ImageClassificationTranslator.Builder) - Constructor for class ai.djl.modality.cv.translator.ImageClassificationTranslator
Constructs an Image Classification using ImageClassificationTranslator.Builder.
ImageClassificationTranslator.Builder - Class in ai.djl.modality.cv.translator
A Builder to construct a ImageClassificationTranslator.
ImageClassificationTranslatorFactory - Class in ai.djl.modality.cv.translator
ImageClassificationTranslatorFactory() - Constructor for class ai.djl.modality.cv.translator.ImageClassificationTranslatorFactory
 
ImageFactory - Class in ai.djl.modality.cv
ImageFactory contains image creation mechanism on top of different platforms like PC and Android.
ImageFactory() - Constructor for class ai.djl.modality.cv.ImageFactory
 
imageHeight - Variable in class ai.djl.modality.cv.translator.ObjectDetectionTranslator
 
imageHeight - Variable in class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder
 
ImageServingTranslator - Class in ai.djl.modality.cv.translator
A Translator that can handle generic CV Input and Output.
ImageServingTranslator(Translator<Image, ?>) - Constructor for class ai.djl.modality.cv.translator.ImageServingTranslator
Constructs a new ImageServingTranslator instance.
imageWidth - Variable in class ai.djl.modality.cv.translator.ObjectDetectionTranslator
 
imageWidth - Variable in class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder
 
includeBias - Variable in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
 
includeBias - Variable in class ai.djl.nn.convolutional.Convolution
 
includeBias - Variable in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
 
includeBias - Variable in class ai.djl.nn.convolutional.Deconvolution
 
includeTokenTypes - Variable in class ai.djl.modality.nlp.translator.QATranslator
 
increment(long) - Method in class ai.djl.training.util.ProgressBar
index - Variable in class ai.djl.training.evaluator.AbstractAccuracy
 
init(Map<String, String>) - Method in class ai.djl.repository.Metadata
Restores artifacts state.
init(String, NDArray[]) - Method in class ai.djl.training.LocalParameterServer
Initializes the ParameterStore for the given parameter.
init(String, NDArray[]) - Method in interface ai.djl.training.ParameterServer
Initializes the ParameterStore for the given parameter.
initialize(NDManager, DataType, Shape...) - Method in class ai.djl.modality.nlp.EncoderDecoder
Initializes the parameters of the block.
initialize(NDManager, DataType, Shape...) - Method in class ai.djl.nn.AbstractBlock
Initializes the parameters of the block.
initialize(NDManager, DataType, Shape...) - Method in interface ai.djl.nn.Block
Initializes the parameters of the block.
initialize(NDManager, DataType) - Method in class ai.djl.nn.Parameter
Initializes the parameter with the given NDManager, with given DataType for the given expected input shapes.
initialize(NDManager, Shape, DataType) - Method in class ai.djl.training.initializer.ConstantInitializer
Initializes a single NDArray.
initialize(NDManager, Shape, DataType) - Method in interface ai.djl.training.initializer.Initializer
Initializes a single NDArray.
initialize(NDManager, Shape, DataType) - Method in class ai.djl.training.initializer.NormalInitializer
Initializes a single NDArray.
initialize(NDManager, Shape, DataType) - Method in class ai.djl.training.initializer.TruncatedNormalInitializer
 
initialize(NDManager, Shape, DataType) - Method in class ai.djl.training.initializer.UniformInitializer
Initializes a single NDArray.
initialize(NDManager, Shape, DataType) - Method in class ai.djl.training.initializer.XavierInitializer
Initializes a single NDArray.
initialize(Shape...) - Method in class ai.djl.training.Trainer
Initializes the Model that the Trainer is going to train.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.modality.nlp.Decoder
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.modality.nlp.embedding.TrainableTextEmbedding
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.modality.nlp.Encoder
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.nn.AbstractBlock
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.nn.ParallelBlock
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.nn.SequentialBlock
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.nn.transformer.BertBlock
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.nn.transformer.BertMaskedLanguageModelBlock
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.nn.transformer.BertNextSentenceBlock
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.nn.transformer.BertPretrainingBlock
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.nn.transformer.IdEmbedding
 
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.nn.transformer.ScaledDotProductAttentionBlock
Initializes the Child blocks of this block.
initializeChildBlocks(NDManager, DataType, Shape...) - Method in class ai.djl.nn.transformer.TransformerEncoderBlock
Initializes the Child blocks of this block.
Initializer - Interface in ai.djl.training.initializer
An interface representing an initialization method.
Input - Class in ai.djl.modality
A class stores the generic input data for inference.
Input() - Constructor for class ai.djl.modality.Input
Constructs a new Input instance.
inputData - Variable in class ai.djl.BaseModel
 
inputForComponent(int, NDList, NDList) - Method in class ai.djl.nn.transformer.BertPretrainingLoss
 
inputForComponent(int, NDList, NDList) - Method in class ai.djl.training.loss.AbstractCompositeLoss
Returns the inputs to computing the loss for a component loss.
inputForComponent(int, NDList, NDList) - Method in class ai.djl.training.loss.SimpleCompositeLoss
Returns the inputs to computing the loss for a component loss.
inputForComponent(int, NDList, NDList) - Method in class ai.djl.training.loss.SingleShotDetectionLoss
Calculate loss between label and prediction.
inputNames - Variable in class ai.djl.nn.AbstractBlock
List of names for the input, named inputs should be manually set in sub class.
inputShape(HpSet) - Method in class ai.djl.training.hyperparameter.EasyHpo
Returns the input shape for the model.
inputShapes - Variable in class ai.djl.nn.AbstractBlock
The shape of the input for this block, set by the initialization process.
InputStreamTranslator<T> - Class in ai.djl.modality.cv.translator.wrapper
Built-in Translator that provides image pre-processing from InputStream.
InputStreamTranslator(Translator<Image, T>) - Constructor for class ai.djl.modality.cv.translator.wrapper.InputStreamTranslator
Creates a UrlTranslator instance.
insert(int, Transform) - Method in class ai.djl.translate.Pipeline
Inserts the given Transform to the list of transforms at the given position.
insert(int, int, Transform) - Method in class ai.djl.translate.Pipeline
Inserts the given Transform to the list of transforms at the given position to be applied on the NDArray at the given index in the input NDList.
insert(int, String, Transform) - Method in class ai.djl.translate.Pipeline
Inserts the given Transform to the list of transforms at the given position to be applied on the NDArray with the given name in the input NDList.
INSTANCE_SEGMENTATION - Static variable in interface ai.djl.Application.CV
An application that finds zero or more objects in an image, the object class (see image classification), and their location as a pixel map.
InstanceSegmentationTranslator - Class in ai.djl.modality.cv.translator
A BaseImageTranslator that post-process the NDArray into DetectedObjects with boundaries at the detailed pixel level.
InstanceSegmentationTranslator(InstanceSegmentationTranslator.Builder) - Constructor for class ai.djl.modality.cv.translator.InstanceSegmentationTranslator
Creates the Instance Segmentation translator from the given builder.
InstanceSegmentationTranslator.Builder - Class in ai.djl.modality.cv.translator
The builder for Instance Segmentation translator.
InstanceSegmentationTranslatorFactory - Class in ai.djl.modality.cv.translator
InstanceSegmentationTranslatorFactory() - Constructor for class ai.djl.modality.cv.translator.InstanceSegmentationTranslatorFactory
 
intern(NDArray) - Method in interface ai.djl.ndarray.NDArray
Replace the handle of the NDArray with the other.
intValue(Map<String, ?>, String) - Static method in class ai.djl.translate.ArgumentsUtil
Returns the integer value from the arguments.
intValue(Map<String, ?>, String, int) - Static method in class ai.djl.translate.ArgumentsUtil
Returns the integer value from the arguments.
inverseTransform(NDArray) - Method in class ai.djl.training.util.MinMaxScaler
Undoes the transformation of X according to feature_range.
inverseTransformi(NDArray) - Method in class ai.djl.training.util.MinMaxScaler
Undoes the transformation of X according to feature_range as an in-place operation.
invoke(String, NDArray[], NDArray[], PairList<String, ?>) - Method in class ai.djl.ndarray.BaseNDManager
An engine specific generic invocation to native operation.
invoke(String, NDList, PairList<String, ?>) - Method in class ai.djl.ndarray.BaseNDManager
An engine specific generic invocation to native operation.
invoke(String, NDArray[], NDArray[], PairList<String, ?>) - Method in interface ai.djl.ndarray.NDManager
An engine specific generic invocation to native operation.
invoke(String, NDList, PairList<String, ?>) - Method in interface ai.djl.ndarray.NDManager
An engine specific generic invocation to native operation.
isArchiveFile(String) - Static method in class ai.djl.repository.FilenameUtils
Returns if the the file is an archive file.
isCHW(Shape) - Static method in class ai.djl.modality.cv.util.NDImageUtils
Check if the shape of the image follows CHW/NCHW.
isClosed - Variable in class ai.djl.ndarray.NDArrayAdapter
 
isControl(char) - Static method in class ai.djl.modality.nlp.NlpUtils
Check whether a character is is considered as a control character.
isDone() - Method in interface ai.djl.modality.rl.env.RlEnv.Step
Returns whether the environment is finished or can accept further actions.
isEmpty() - Method in interface ai.djl.ndarray.NDArray
Returns true if this NDArray is special case: no-value NDArray.
isFloating() - Method in enum ai.djl.ndarray.types.DataType
Checks whether it is a floating data type.
isGpu() - Method in class ai.djl.Device
Returns if the Device is GPU.
isHyphenLike(Integer) - Static method in class ai.djl.modality.nlp.preprocess.HyphenNormalizer
Returns whether the given code point is a hyphen-like codepoint.
isInfinite() - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray with value true where this NDArray's entries are infinite, or false where they are not infinite.
isInfinite() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray with value true where this NDArray's entries are infinite, or false where they are not infinite.
isInitialized() - Method in class ai.djl.nn.AbstractBlock
Returns a boolean whether the block is initialized.
isInitialized() - Method in interface ai.djl.nn.Block
Returns a boolean whether the block is initialized.
isInitialized() - Method in class ai.djl.nn.Parameter
Checks if this Parameter is initialized.
isInteger() - Method in enum ai.djl.ndarray.types.DataType
Checks whether it is an integer data type.
isLayoutKnown() - Method in class ai.djl.ndarray.types.Shape
Returns true if a layout is set.
isNaN() - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray with value true where this NDArray's entries are NaN, or false where they are not NaN.
isNaN() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray with value true where this NDArray's entries are NaN, or false where they are not NaN.
isOpen() - Method in class ai.djl.ndarray.BaseNDManager
Check if the manager is still valid.
isOpen() - Method in interface ai.djl.ndarray.NDManager
Check if the manager is still valid.
isPunctuation(char) - Static method in class ai.djl.modality.nlp.NlpUtils
Check whether a character is considered as a punctuation.
isRemote() - Method in class ai.djl.repository.JarRepository
Returns whether the repository is remote repository.
isRemote() - Method in class ai.djl.repository.LocalRepository
Returns whether the repository is remote repository.
isRemote() - Method in class ai.djl.repository.RemoteRepository
Returns whether the repository is remote repository.
isRemote() - Method in interface ai.djl.repository.Repository
Returns whether the repository is remote repository.
isRemote() - Method in class ai.djl.repository.SimpleRepository
Returns whether the repository is remote repository.
isRemote() - Method in class ai.djl.repository.SimpleUrlRepository
Returns whether the repository is remote repository.
isScalar() - Method in interface ai.djl.ndarray.NDArray
Returns true if this NDArray is a scalar NDArray with empty Shape.
isScalar() - Method in class ai.djl.ndarray.types.Shape
Returns true if the NDArray is a scalar.
isSnapshot() - Method in class ai.djl.repository.Artifact
Returns true if the artifact is a snapshot.
isSnapshot() - Method in class ai.djl.repository.Version
Returns true if this is a snapshot version.
isSparse() - Method in interface ai.djl.ndarray.NDArray
Returns true if this NDArray is a SparseNDArray.
isSupported(Class<?>, Class<?>) - Method in class ai.djl.translate.DefaultTranslatorFactory
Returns if the input/output is supported by the TranslatorFactory.
isSupported(Class<?>, Class<?>) - Method in interface ai.djl.translate.TranslatorFactory
Returns if the input/output is supported by the TranslatorFactory.
isWhiteSpace(char) - Static method in class ai.djl.modality.nlp.NlpUtils
Check whether a character is is considered as a whitespace.
item(int) - Method in class ai.djl.modality.Classifications
Returns the item at a given index based on the order used to construct the Classifications.
item(int) - Method in class ai.djl.modality.cv.output.DetectedObjects
Returns the item at a given index based on the order used to construct the Classifications.
Item() - Constructor for class ai.djl.repository.Artifact.Item
 
items() - Method in class ai.djl.modality.Classifications
Returns a classification item for each potential class for the input.
iterateDataset(Dataset) - Method in class ai.djl.training.Trainer
Fetches an iterator that can iterate through the given Dataset.
iterator() - Method in class ai.djl.training.dataset.DataIterable

J

JarRepository - Class in ai.djl.repository
A JarRepository is a Repository contains an archive file from classpath.
Joint(double, double, double) - Constructor for class ai.djl.modality.cv.output.Joints.Joint
Constructs a Joint with given data.
Joints - Class in ai.djl.modality.cv.output
A result of all joints found during Human Pose Estimation on a single image.
Joints(List<Joints.Joint>) - Constructor for class ai.djl.modality.cv.output.Joints
Constructs the Joints with the provided joints.
Joints.Joint - Class in ai.djl.modality.cv.output
A joint that was detected using Human Pose Estimation on an image.

K

kernelShape - Variable in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
 
kernelShape - Variable in class ai.djl.nn.convolutional.Convolution
 
kernelShape - Variable in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
 
kernelShape - Variable in class ai.djl.nn.convolutional.Deconvolution
 

L

L1Loss - Class in ai.djl.training.loss
L1Loss calculates L1 loss between label and prediction.
L1Loss() - Constructor for class ai.djl.training.loss.L1Loss
Calculates L1 Loss between the label and prediction, a.k.a.
L1Loss(String) - Constructor for class ai.djl.training.loss.L1Loss
Calculates L1 Loss between the label and prediction, a.k.a.
L1Loss(String, float) - Constructor for class ai.djl.training.loss.L1Loss
Calculates L1 Loss between the label and prediction, a.k.a.
l1Loss() - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L1Loss with default weight and batch axis.
l1Loss(String) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L1Loss with default weight and batch axis.
l1Loss(String, float) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L1Loss with given weight and batch axis.
L1WeightDecay - Class in ai.djl.training.loss
L1WeightDecay calculates L1 penalty of a set of parameters.
L1WeightDecay(NDList) - Constructor for class ai.djl.training.loss.L1WeightDecay
Calculates L1 weight decay for regularization.
L1WeightDecay(String, NDList) - Constructor for class ai.djl.training.loss.L1WeightDecay
Calculates L1 weight decay for regularization.
L1WeightDecay(String, NDList, float) - Constructor for class ai.djl.training.loss.L1WeightDecay
Calculates L1 weight decay for regularization.
l1WeightedDecay(NDList) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L1WeightDecay with default weight and name.
l1WeightedDecay(String, NDList) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L1WeightDecay with default weight.
l1WeightedDecay(String, float, NDList) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L1WeightDecay.
L2Loss - Class in ai.djl.training.loss
Calculates L2Loss between label and prediction, a.k.a.
L2Loss() - Constructor for class ai.djl.training.loss.L2Loss
Calculate L2Loss between the label and prediction, a.k.a.
L2Loss(String) - Constructor for class ai.djl.training.loss.L2Loss
Calculate L2Loss between the label and prediction, a.k.a.
L2Loss(String, float) - Constructor for class ai.djl.training.loss.L2Loss
Calculates L2Loss between the label and prediction, a.k.a.
l2Loss() - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L2Loss with default weight and batch axis.
l2Loss(String) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L2Loss with default weight and batch axis.
l2Loss(String, float) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L2Loss with given weight and batch axis.
L2WeightDecay - Class in ai.djl.training.loss
L2WeightDecay calculates L2 penalty of a set of parameters.
L2WeightDecay(NDList) - Constructor for class ai.djl.training.loss.L2WeightDecay
Calculates L2 weight decay for regularization.
L2WeightDecay(String, NDList) - Constructor for class ai.djl.training.loss.L2WeightDecay
Calculates L2 weight decay for regularization.
L2WeightDecay(String, NDList, float) - Constructor for class ai.djl.training.loss.L2WeightDecay
Calculates L2 weight decay for regularization.
l2WeightedDecay(NDList) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L2WeightDecay with default weight and name.
l2WeightedDecay(String, NDList) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L2WeightDecay with default weight.
l2WeightedDecay(String, float, NDList) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of L2WeightDecay.
labelBatchifier - Variable in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
 
labelBatchifier - Variable in class ai.djl.training.dataset.RandomAccessDataset
 
labels - Variable in class ai.djl.training.dataset.ArrayDataset
 
LambdaBlock - Class in ai.djl.nn
LambdaBlock is a Block with no parameters or children.
LambdaBlock(Function<NDList, NDList>) - Constructor for class ai.djl.nn.LambdaBlock
Creates a LambdaBlock that can apply the specified function.
LambdaProcessor - Class in ai.djl.modality.nlp.preprocess
TextProcessor will apply user defined lambda function on input tokens.
LambdaProcessor(Function<String, String>) - Constructor for class ai.djl.modality.nlp.preprocess.LambdaProcessor
Creates a LambdaProcessor and specify the function to apply.
Landmark - Class in ai.djl.modality.cv.output
Landmark is the container that stores the key points for landmark on a single face.
Landmark(double, double, double, double, List<Point>) - Constructor for class ai.djl.modality.cv.output.Landmark
Constructs a Landmark using a list of points.
large() - Method in class ai.djl.nn.transformer.BertBlock.Builder
Sets this builder's params to the LARGE config of the original BERT paper.
latestMetric(String) - Method in class ai.djl.metric.Metrics
Returns the latest Metric with the specified metric name.
LayerNorm - Class in ai.djl.nn.norm
Layer normalization works by normalizing the values of input data for each input sample to have mean of 0 and variance of 1.
layerNorm(NDArray, Shape, NDArray, NDArray, float) - Static method in class ai.djl.nn.norm.LayerNorm
Applies Layer Normalization with average and variance for each input sample across the axis dimensions.
LayerNorm.Builder - Class in ai.djl.nn.norm
The Builder to construct a LayerNorm.
LayoutType - Enum in ai.djl.ndarray.types
An enum to represent the meaning of a particular axis in an NDArray.
LazyNDArray - Interface in ai.djl.ndarray
An NDArray that waits to compute values until they are needed.
leakyRelu(NDArray, float) - Static method in class ai.djl.nn.Activation
Applies Leaky ReLU activation on the input NDArray.
leakyRelu(NDList, float) - Static method in class ai.djl.nn.Activation
Applies Leaky ReLU activation on the input singleton NDList.
leakyReluBlock(float) - Static method in class ai.djl.nn.Activation
Creates a LambdaBlock that applies the LeakyReLU activation function in its forward function.
License - Class in ai.djl.repository
A License is a container to save the license information.
License() - Constructor for class ai.djl.repository.License
 
licenses - Variable in class ai.djl.repository.Metadata
 
like() - Method in interface ai.djl.ndarray.NDArray
Returns an uninitialized NDArray with the same Shape, DataType and SparseFormat as the input NDArray.
limit - Variable in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
 
limit - Variable in class ai.djl.training.dataset.RandomAccessDataset
 
Linear - Class in ai.djl.nn.core
A Linear block applies a linear transformation \(Y = XW^T + b\).
linear(NDArray, NDArray) - Static method in class ai.djl.nn.core.Linear
Applies a linear transformation to the incoming data.
linear(NDArray, NDArray, NDArray) - Static method in class ai.djl.nn.core.Linear
Applies a linear transformation to the incoming data.
Linear.Builder - Class in ai.djl.nn.core
The Builder to construct a Linear type of Block.
LINEAR_REGRESSION - Static variable in interface ai.djl.Application.Tabular
An application that takes a feature vector (table row) and predicts a numerical feature based on it.
LinearTracker - Class in ai.djl.training.tracker
FactorTracker is an implementation of Tracker which is updated by a constant factor.
LinearTracker(LinearTracker.Builder) - Constructor for class ai.djl.training.tracker.LinearTracker
Creates a new instance of FactorTracker.
LinearTracker.Builder - Class in ai.djl.training.tracker
The Builder to construct an LinearTracker object.
linspace(float, float, int, boolean) - Method in class ai.djl.ndarray.BaseNDManager
Returns evenly spaced numbers over a specified interval.
linspace(int, int, int) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced numbers over a specified interval.
linspace(float, float, int) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced numbers over a specified interval.
linspace(int, int, int, boolean) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced numbers over a specified interval.
linspace(float, float, int, boolean) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced numbers over a specified interval.
linspace(float, float, int, boolean, Device) - Method in interface ai.djl.ndarray.NDManager
Returns evenly spaced numbers over a specified interval.
listArtifacts() - Method in class ai.djl.repository.MRL
Returns a list of artifacts in this resource.
listDirectory(Artifact.Item, String) - Method in class ai.djl.repository.AbstractRepository
Returns the list of files directly within a specified directory in a zipped directory item.
listDirectory(Artifact.Item, String) - Method in interface ai.djl.repository.Repository
Returns the list of files directly within a specified directory in a zipped directory item.
listModels() - Method in class ai.djl.repository.zoo.BaseModelLoader
Returns a list of the available artifacts that can be loaded.
listModels() - Method in interface ai.djl.repository.zoo.ModelLoader
Returns a list of the available artifacts that can be loaded.
listModels() - Static method in class ai.djl.repository.zoo.ModelZoo
Returns the available Application and their model artifact metadata.
listModels(Criteria<?, ?>) - Static method in class ai.djl.repository.zoo.ModelZoo
Returns the available Application and their model artifact metadata.
listModelZoo() - Static method in class ai.djl.repository.zoo.ModelZoo
Returns available model zoos.
load(Model) - Method in class ai.djl.modality.cv.translator.BaseImageTranslator.SynsetLoader
 
load(Path) - Method in interface ai.djl.Model
Loads the model from the modelPath.
load(Path, String) - Method in interface ai.djl.Model
Loads the model from the modelPath and the given name.
load(Path, String, Map<String, ?>) - Method in interface ai.djl.Model
Loads the model from the modelPath with the name and options provided.
load(Path) - Method in class ai.djl.ndarray.BaseNDManager
Loads the NDArrays saved to a file.
load(Path) - Method in interface ai.djl.ndarray.NDManager
Loads the NDArrays saved to a file.
load(Path, Device) - Method in interface ai.djl.ndarray.NDManager
Loads the NDArrays saved to a file.
load(NDManager, DataInputStream) - Method in class ai.djl.nn.Parameter
Loads parameter NDArrays from InputStream.
load(Path, String, Map<String, ?>) - Method in class ai.djl.repository.zoo.ZooModel
Loads the model from the modelPath with the name and options provided.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.AbstractBlock
Overwrite this to load additional metadata with the parameter values.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.convolutional.Convolution
Overwrite this to load additional metadata with the parameter values.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.convolutional.Deconvolution
Overwrite this to load additional metadata with the parameter values.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.core.Linear
Overwrite this to load additional metadata with the parameter values.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.core.Prelu
Overwrite this to load additional metadata with the parameter values.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.norm.BatchNorm
Overwrite this to load additional metadata with the parameter values.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.norm.Dropout
Overwrite this to load additional metadata with the parameter values.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.norm.LayerNorm
Overwrite this to load additional metadata with the parameter values.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.ParallelBlock
Overwrite this to load additional metadata with the parameter values.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.recurrent.RecurrentBlock
Overwrite this to load additional metadata with the parameter values.
loadMetadata(byte, DataInputStream) - Method in class ai.djl.nn.SequentialBlock
Overwrite this to load additional metadata with the parameter values.
loadModel(Criteria<I, O>) - Method in class ai.djl.repository.zoo.BaseModelLoader
Loads the model with the given criteria.
loadModel() - Method in class ai.djl.repository.zoo.Criteria
Load the ZooModel that matches this criteria.
loadModel(Criteria<I, O>) - Method in interface ai.djl.repository.zoo.ModelLoader
Loads the model with the given criteria.
loadModel(Criteria<I, O>) - Static method in class ai.djl.repository.zoo.ModelZoo
Load the ZooModel that matches this criteria.
loadParameters(NDManager, DataInputStream) - Method in class ai.djl.modality.nlp.Decoder
Loads the parameters from the given input stream.
loadParameters(NDManager, DataInputStream) - Method in class ai.djl.modality.nlp.Encoder
Loads the parameters from the given input stream.
loadParameters(NDManager, DataInputStream) - Method in class ai.djl.modality.nlp.EncoderDecoder
Loads the parameters from the given input stream.
loadParameters(NDManager, DataInputStream) - Method in class ai.djl.nn.AbstractBlock
Loads the parameters from the given input stream.
loadParameters(NDManager, DataInputStream) - Method in interface ai.djl.nn.Block
Loads the parameters from the given input stream.
loadParameters(NDManager, DataInputStream) - Method in class ai.djl.nn.core.ConstantEmbedding
Loads the parameters from the given input stream.
loadParameters(NDManager, DataInputStream) - Method in class ai.djl.nn.core.Embedding
Loads the parameters from the given input stream.
loadParameters(NDManager, DataInputStream) - Method in class ai.djl.nn.LambdaBlock
Loads the parameters from the given input stream.
locale - Variable in class ai.djl.modality.nlp.translator.QATranslator
 
LocalParameterServer - Class in ai.djl.training
LocalParameterServer is an implementation of the ParameterServer interface.
LocalParameterServer(Optimizer) - Constructor for class ai.djl.training.LocalParameterServer
Create a new instance of LocalParameterServer for the given optimizer.
LocalRepository - Class in ai.djl.repository
A LocalRepository is a Repository located in a filesystem directory.
LocalRepository(String, URI, Path) - Constructor for class ai.djl.repository.LocalRepository
(Internal) Constructs a LocalRepository from the path with inferred name.
locate(MRL) - Method in class ai.djl.repository.JarRepository
Returns the metadata at a mrl.
locate(MRL) - Method in class ai.djl.repository.LocalRepository
Returns the metadata at a mrl.
locate(MRL) - Method in class ai.djl.repository.RemoteRepository
Returns the metadata at a mrl.
locate(MRL) - Method in interface ai.djl.repository.Repository
Returns the metadata at a mrl.
locate(MRL) - Method in class ai.djl.repository.SimpleRepository
Returns the metadata at a mrl.
locate(MRL) - Method in class ai.djl.repository.SimpleUrlRepository
Returns the metadata at a mrl.
log() - Method in interface ai.djl.ndarray.NDArray
Returns the natural logarithmic value of this NDArray element-wise.
log() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the natural logarithmic value of this NDArray element-wise.
log10() - Method in interface ai.djl.ndarray.NDArray
Returns the base 10 logarithm of this NDArray element-wise.
log10() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the base 10 logarithm of this NDArray element-wise.
log2() - Method in interface ai.djl.ndarray.NDArray
Returns the base 2 logarithm of this NDArray element-wise.
log2() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the base 2 logarithm of this NDArray element-wise.
logging() - Static method in interface ai.djl.training.listener.TrainingListener.Defaults
A default TrainingListener set including batch output logging.
logging(int) - Static method in interface ai.djl.training.listener.TrainingListener.Defaults
A default TrainingListener set including batch output logging.
logging(String) - Static method in interface ai.djl.training.listener.TrainingListener.Defaults
A default TrainingListener set including batch output logging and output directory.
LoggingTrainingListener - Class in ai.djl.training.listener
TrainingListener that outputs the progress of training each batch and epoch into logs.
LoggingTrainingListener() - Constructor for class ai.djl.training.listener.LoggingTrainingListener
Constructs a LoggingTrainingListener instance.
LoggingTrainingListener(int) - Constructor for class ai.djl.training.listener.LoggingTrainingListener
Constructs a LoggingTrainingListener instance with specified steps.
logicalAnd(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns the truth value of this NDArray AND the other NDArray element-wise.
logicalAnd(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the truth value of this NDArray AND the other NDArray element-wise.
logicalAnd(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the truth value of NDArray a AND NDArray b element-wise.
logicalNot() - Method in interface ai.djl.ndarray.NDArray
Computes the truth value of NOT this NDArray element-wise.
logicalNot() - Method in class ai.djl.ndarray.NDArrayAdapter
Computes the truth value of NOT this NDArray element-wise.
logicalOr(NDArray) - Method in interface ai.djl.ndarray.NDArray
Computes the truth value of this NDArray OR the other NDArray element-wise.
logicalOr(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Computes the truth value of this NDArray OR the other NDArray element-wise.
logicalOr(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Computes the truth value of NDArray a AND NDArray b element-wise.
logicalXor(NDArray) - Method in interface ai.djl.ndarray.NDArray
Computes the truth value of this NDArray XOR the other NDArray element-wise.
logicalXor(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Computes the truth value of this NDArray XOR the other NDArray element-wise.
logicalXor(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Computes the truth value of NDArray a AND NDArray b element-wise.
logSoftmax(int) - Method in interface ai.djl.ndarray.NDArray
Applies the softmax function followed by a logarithm.
logSoftmax(int) - Method in class ai.djl.ndarray.NDArrayAdapter
Applies the softmax function followed by a logarithm.
longValue(Map<String, ?>, String) - Static method in class ai.djl.translate.ArgumentsUtil
Returns the long value from the arguments.
longValue(Map<String, ?>, String, long) - Static method in class ai.djl.translate.ArgumentsUtil
Returns the long value from the arguments.
Loss - Class in ai.djl.training.loss
Loss functions (or Cost functions) are used to evaluate the model predictions against true labels for optimization.
Loss(String) - Constructor for class ai.djl.training.loss.Loss
Base class for metric with abstract update methods.
LowerCaseConvertor - Class in ai.djl.modality.nlp.preprocess
LowerCaseConvertor converts every character of the input tokens to it's respective lower case character.
LowerCaseConvertor(Locale) - Constructor for class ai.djl.modality.nlp.preprocess.LowerCaseConvertor
Creates a TextProcessor that converts input text into lower case character given the Locale.
LowerCaseConvertor() - Constructor for class ai.djl.modality.nlp.preprocess.LowerCaseConvertor
Creates a TextProcessor that converts input text into lower case character with the default english Locale.
lpPool1d(NDArray, float, Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Performs 1-D LP Pooling on the input.
lpPool1dBlock(float, Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the lpPool1dBlock pooling function in its forward function.
lpPool1dBlock(float, Shape, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the lpPool1dBlock pooling function in its forward function.
lpPool1dBlock(float, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the lpPool1dBlock pooling function in its forward function.
lpPool2d(NDArray, float, Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Performs 2-D LP Pooling on the input.
lpPool2dBlock(float, Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the lpPool2dBlock pooling function in its forward function.
lpPool2dBlock(float, Shape, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the lpPool2dBlock pooling function in its forward function.
lpPool2dBlock(float, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the lpPool2dBlock pooling function in its forward function.
lpPool2dBlock(float, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the lpPool2dBlock pooling function in its forward function.
lpPool3d(NDArray, float, Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Performs 3-D LP Pooling on the input.
lpPool3dBlock(float, Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the lpPool3dBlock pooling function in its forward function.
lpPool3dBlock(float, Shape, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the lpPool3dBlock pooling function in its forward function.
lpPool3dBlock(float, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the LpPoo3D pooling function in its forward function.
lpPool3dBlock(float, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the lpPool3dBlock pooling function in its forward function.
LruReplayBuffer - Class in ai.djl.modality.rl
A simple ReplayBuffer that randomly selects across the whole buffer, but always removes the oldest items in the buffer once it is full.
LruReplayBuffer(int, int) - Constructor for class ai.djl.modality.rl.LruReplayBuffer
Constructs a LruReplayBuffer.
LSTM - Class in ai.djl.nn.recurrent
LSTM is an implementation of recurrent neural networks which applies Long Short-Term Memory recurrent layer to input.
LSTM.Builder - Class in ai.djl.nn.recurrent
The Builder to construct a LSTM type of Block.
lt(Number) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Less" comparison.
lt(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Less" comparison.
lt(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Less" comparison.
lt(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Less" comparison.
lt(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Less" comparison.
lt(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Less" comparison.
lt(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Less" comparison.
lte(Number) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Less or equals" comparison.
lte(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Less or equals" comparison.
lte(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Less or equals" comparison.
lte(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Less or equals" comparison.
lte(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Less or equals" comparison.
lte(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Less or equals" comparison.
lte(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Less or equals" comparison.

M

MACHINE_TRANSLATION - Static variable in interface ai.djl.Application.NLP
An application that translates text from one language to another.
MalformedModelException - Exception in ai.djl
Thrown to indicate Model parameters are not in expected format or are malformed.
MalformedModelException(String) - Constructor for exception ai.djl.MalformedModelException
Constructs a new exception with the specified detail message.
MalformedModelException(String, Throwable) - Constructor for exception ai.djl.MalformedModelException
Constructs a new exception with the specified detail message and cause.
MalformedModelException(Throwable) - Constructor for exception ai.djl.MalformedModelException
Constructs a new exception with the specified cause and a detail message of (cause==null ? null : cause.toString()) which typically contains the class and detail message of cause.
manager - Variable in class ai.djl.BaseModel
 
manager - Variable in class ai.djl.inference.Predictor
 
manager - Variable in class ai.djl.ndarray.NDArrayAdapter
 
map(Function<Pair<Long, LayoutType>, Pair<Long, LayoutType>>) - Method in class ai.djl.ndarray.types.Shape
Returns a mapped shape.
Mask - Class in ai.djl.modality.cv.output
A mask with a probability for each pixel within a bounding rectangle.
Mask(double, double, double, double, float[][]) - Constructor for class ai.djl.modality.cv.output.Mask
Constructs a Mask with the given data.
maskedSoftmaxCrossEntropyLoss() - Static method in class ai.djl.training.loss.Loss
Returns a new instance of MaskedSoftmaxCrossEntropyLoss with default arguments.
maskedSoftmaxCrossEntropyLoss(String) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of MaskedSoftmaxCrossEntropyLoss with default arguments.
maskedSoftmaxCrossEntropyLoss(String, float, int, boolean, boolean) - Static method in class ai.djl.training.loss.Loss
Returns a new instance of MaskedSoftmaxCrossEntropyLoss with the given arguments.
MaskedSoftmaxCrossEntropyLoss - Class in ai.djl.training.loss
MaskedSoftmaxCrossEntropyLoss is an implementation of Loss that only considers a specific number of values for the loss computations, and masks the rest according to the given sequence.
MaskedSoftmaxCrossEntropyLoss() - Constructor for class ai.djl.training.loss.MaskedSoftmaxCrossEntropyLoss
Creates a new instance of SoftmaxCrossEntropyLoss with default parameters.
MaskedSoftmaxCrossEntropyLoss(String) - Constructor for class ai.djl.training.loss.MaskedSoftmaxCrossEntropyLoss
Creates a new instance of SoftmaxCrossEntropyLoss with default parameters.
MaskedSoftmaxCrossEntropyLoss(String, float, int, boolean, boolean) - Constructor for class ai.djl.training.loss.MaskedSoftmaxCrossEntropyLoss
Creates a new instance of MaskedSoftmaxCrossEntropyLoss with the given parameters.
match(Map<String, String>) - Method in class ai.djl.repository.MRL
Returns the first artifact that matches a given criteria.
MatchAllMetadata() - Constructor for class ai.djl.repository.Metadata.MatchAllMetadata
Creates a MatchAllMetadata instance.
matches(Application) - Method in class ai.djl.Application
Returns whether this application matches the test application set.
matches(List<Artifact>) - Method in class ai.djl.repository.VersionRange
Filters the provided artifacts to those that match the version range.
matMul(NDArray) - Method in interface ai.djl.ndarray.NDArray
Product matrix of this NDArray and the other NDArray.
matMul(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Product matrix of this NDArray and the other NDArray.
matMul(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Product matrix of this NDArray and the other NDArray.
max() - Method in interface ai.djl.ndarray.NDArray
Returns the maximum of this NDArray.
max(int[]) - Method in interface ai.djl.ndarray.NDArray
Returns the maximum of this NDArray along given axes.
max(int[], boolean) - Method in interface ai.djl.ndarray.NDArray
Returns the maximum of this NDArray along given axes.
max() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the maximum of this NDArray.
max(int[], boolean) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the maximum of this NDArray along given axes.
maximum(Number) - Method in interface ai.djl.ndarray.NDArray
Returns the maximum of this NDArray and a number element-wise.
maximum(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns the maximum of this NDArray and the other NDArray element-wise.
maximum(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the maximum of this NDArray and a number element-wise.
maximum(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the maximum of this NDArray and the other NDArray element-wise.
maximum(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns the maximum of a NDArray and a number element-wise.
maximum(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the maximum of a number and a NDArray element-wise.
maximum(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the maximum of NDArray a and NDArray b element-wise.
maxLabels - Variable in class ai.djl.modality.nlp.translator.QATranslator
 
maxLength - Variable in class ai.djl.modality.nlp.translator.QATranslator
 
maxPool1d(NDArray, Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Performs 1-D Max Pooling on the input.
maxPool1dBlock(Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool1d pooling function in its forward function.
maxPool1dBlock(Shape, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool1dBlock pooling function in its forward function.
maxPool1dBlock(Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool1dBlock pooling function in its forward function.
maxPool1dBlock(Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool1dBlock pooling function in its forward function.
maxPool2d(NDArray, Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Performs 2-D Max Pooling on the input.
maxPool2dBlock(Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool2dBlock pooling function in its forward function.
maxPool2dBlock(Shape, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool2dBlock pooling function in its forward function.
maxPool2dBlock(Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool2dBlock pooling function in its forward function.
maxPool2dBlock(Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool2dBlock pooling function in its forward function.
maxPool3d(NDArray, Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Performs 3-D Max Pooling on the input.
maxPool3dBlock(Shape, Shape, Shape, boolean) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool3dBlock pooling function in its forward function.
maxPool3dBlock(Shape, Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool3dBlock pooling function in its forward function.
maxPool3dBlock(Shape, Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool3dBlock pooling function in its forward function.
maxPool3dBlock(Shape) - Static method in class ai.djl.nn.pooling.Pool
Creates a LambdaBlock that applies the maxPool3dBlock pooling function in its forward function.
md5hash(String) - Static method in class ai.djl.repository.AbstractRepository
 
mean(String) - Method in class ai.djl.metric.Metrics
Returns the average value of the specified metric.
mean() - Method in interface ai.djl.ndarray.NDArray
Returns the average of this NDArray.
mean(int[]) - Method in interface ai.djl.ndarray.NDArray
Returns the average of this NDArray along given axes.
mean(int[], boolean) - Method in interface ai.djl.ndarray.NDArray
Returns the average of this NDArray along given axes.
mean() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the average of this NDArray.
mean(int[], boolean) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the average of this NDArray along given axes.
median() - Method in interface ai.djl.ndarray.NDArray
Returns median value for this NDArray.
median(int[]) - Method in interface ai.djl.ndarray.NDArray
Returns median value along given axes.
median() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns median value for this NDArray.
median(int[]) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns median value along given axes.
MemoryTrainingListener - Class in ai.djl.training.listener
TrainingListener that collects the memory usage information.
MemoryTrainingListener() - Constructor for class ai.djl.training.listener.MemoryTrainingListener
Constructs a MemoryTrainingListener that does not output data to a file.
MemoryTrainingListener(String) - Constructor for class ai.djl.training.listener.MemoryTrainingListener
Constructs a MemoryTrainingListener that outputs data in the given directory.
Metadata - Class in ai.djl.repository
A Metadata is a collection of Artifacts with unified metadata (including MRL) that are stored in the same "metadata.json" file.
Metadata() - Constructor for class ai.djl.repository.Metadata
 
Metadata.MatchAllMetadata - Class in ai.djl.repository
A Metadata class that matches all any search criteria.
Metric - Class in ai.djl.metric
A class representing a single recorded Metric value.
Metric(String, Number) - Constructor for class ai.djl.metric.Metric
Constructs a Metric instance with the specified metricName and value.
Metric(String, Number, String) - Constructor for class ai.djl.metric.Metric
Constructs a Metric instance with the specified metricName, value , and unit.
metricName(Evaluator, String) - Static method in class ai.djl.training.listener.EvaluatorTrainingListener
Returns the metric created with the evaluator for the given stage.
metrics - Variable in class ai.djl.inference.Predictor
 
Metrics - Class in ai.djl.metric
A collection of Metric objects organized by metric name.
Metrics() - Constructor for class ai.djl.metric.Metrics
Constructs an empty Metrics instance.
micro() - Method in class ai.djl.nn.transformer.BertBlock.Builder
Sets this builder's params to a minimal configuration that nevertheless performs quite well.
min() - Method in interface ai.djl.ndarray.NDArray
Returns the minimum of this NDArray.
min(int[]) - Method in interface ai.djl.ndarray.NDArray
Returns the minimum of this NDArray along given axes.
min(int[], boolean) - Method in interface ai.djl.ndarray.NDArray
Returns the minimum of this NDArray along given axes.
min() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the minimum of this NDArray.
min(int[], boolean) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the minimum of this NDArray along given axes.
minimum(Number) - Method in interface ai.djl.ndarray.NDArray
Returns the minimum of this NDArray and a number element-wise.
minimum(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns the minimum of this NDArray and the other NDArray element-wise.
minimum(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the minimum of this NDArray and a number element-wise.
minimum(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the minimum of this NDArray and the other NDArray element-wise.
minimum(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns the minimum of a NDArray and a number element-wise.
minimum(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the minimum of a number and a NDArray element-wise.
minimum(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the minimum of NDArray a and NDArray b element-wise.
MinMaxScaler - Class in ai.djl.training.util
Transform arrays by scaling each value to a given range.
MinMaxScaler() - Constructor for class ai.djl.training.util.MinMaxScaler
 
mish(NDArray) - Static method in class ai.djl.nn.Activation
Applies Mish activation on the input NDArray.
mish(NDList) - Static method in class ai.djl.nn.Activation
Applies Mish activation on the input singleton NDList.
mishBlock() - Static method in class ai.djl.nn.Activation
Creates a LambdaBlock that applies the Mish activation function in its forward function.
MissingOps - Class in ai.djl.nn.transformer
Operators missing from NDArray that are necessary to implement Bert pretraining.
MKL - Static variable in class ai.djl.engine.StandardCapabilities
 
MKLDNN - Static variable in class ai.djl.engine.StandardCapabilities
 
mod(Number) - Method in interface ai.djl.ndarray.NDArray
Returns element-wise remainder of division.
mod(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns element-wise remainder of division.
mod(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns element-wise remainder of division.
mod(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns element-wise remainder of division.
mod(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns element-wise remainder of division.
mod(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns element-wise remainder of division.
mod(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns element-wise remainder of division.
Model - Interface in ai.djl
A model is a collection of artifacts that is created by the training process.
model(Repository, Application, String, String, String, String) - Static method in class ai.djl.repository.MRL
Creates a model MRL with specified application.
model(Application, String, String) - Method in interface ai.djl.repository.Repository
Creates a model MRL with specified application.
model(Application, String, String, String) - Method in interface ai.djl.repository.Repository
Creates a model MRL with specified application.
model(Application, String, String, String, String) - Method in interface ai.djl.repository.Repository
Creates a model MRL with specified application.
modelDir - Variable in class ai.djl.BaseModel
 
ModelException - Exception in ai.djl
Thrown to indicate Model parameter or load exceptions parent to Model Exceptions.
ModelException(String) - Constructor for exception ai.djl.ModelException
Constructs a new exception with the specified detail message.
ModelException(String, Throwable) - Constructor for exception ai.djl.ModelException
Constructs a new exception with the specified detail message and cause.
ModelException(Throwable) - Constructor for exception ai.djl.ModelException
Constructs a new exception with the specified cause and a detail message of (cause==null ? null : cause.toString()) which typically contains the class and detail message of cause.
ModelLoader - Interface in ai.djl.repository.zoo
A ModelLoader loads a particular ZooModel from a Repository for a model zoo.
modelName - Variable in class ai.djl.BaseModel
 
ModelNotFoundException - Exception in ai.djl.repository.zoo
Thrown when an application tries to load a model from repository search path.
ModelNotFoundException(String) - Constructor for exception ai.djl.repository.zoo.ModelNotFoundException
Constructs a new exception with the specified detail message.
ModelNotFoundException(String, Throwable) - Constructor for exception ai.djl.repository.zoo.ModelNotFoundException
Constructs a new exception with the specified detail message and cause.
ModelNotFoundException(Throwable) - Constructor for exception ai.djl.repository.zoo.ModelNotFoundException
Constructs a new exception with the specified cause and a detail message of (cause==null ? null : cause.toString()) (which typically contains the class and detail message of cause).
ModelZoo - Class in ai.djl.repository.zoo
An interface represents a collection of models.
ModelZoo() - Constructor for class ai.djl.repository.zoo.ModelZoo
 
ModelZooTextEmbedding - Class in ai.djl.modality.nlp.embedding
ModelZooTextEmbedding(Model) - Constructor for class ai.djl.modality.nlp.embedding.ModelZooTextEmbedding
Constructs a ModelZooTextEmbedding.
modi(Number) - Method in interface ai.djl.ndarray.NDArray
Returns element-wise remainder of division in place.
modi(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns in place element-wise remainder of division in place.
modi(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns element-wise remainder of division in place.
modi(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns in place element-wise remainder of division in place.
modi(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns element-wise remainder of division in place.
modi(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns element-wise remainder of division in place.
modi(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns element-wise remainder of division.
MRL - Class in ai.djl.repository
The MRL (Machine learning Resource Locator) is a pointer to a Metadata "resource" on a machine learning Repository.
mrl - Variable in class ai.djl.repository.zoo.BaseModelLoader
 
mul(Number) - Method in interface ai.djl.ndarray.NDArray
Multiplies this NDArray by a number element-wise.
mul(NDArray) - Method in interface ai.djl.ndarray.NDArray
Multiplies this NDArray by other NDArrays element-wise.
mul(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Multiplies this NDArray by a number element-wise.
mul(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Multiplies this NDArray by other NDArrays element-wise.
mul(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Multiplies the NDArray by a number element-wise.
mul(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Multiplies a number by a NDArray element-wise.
mul(NDArray...) - Static method in class ai.djl.ndarray.NDArrays
Multiplies all of the NDArrays together element-wise.
muli(Number) - Method in interface ai.djl.ndarray.NDArray
Multiplies this NDArray by a number element-wise in place.
muli(NDArray) - Method in interface ai.djl.ndarray.NDArray
Multiplies this NDArray by other NDArray element-wise in place.
muli(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Multiplies this NDArray by a number element-wise in place.
muli(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Multiplies this NDArray by other NDArray element-wise in place.
muli(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Multiplies the NDArray by a number element-wise in place.
muli(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Multiplies a number by a NDArray element-wise.
muli(NDArray...) - Static method in class ai.djl.ndarray.NDArrays
Multiplies all of the NDArrays together element-wise in place.
MultiBoxDetection - Class in ai.djl.modality.cv
MultiBoxDetection is the class that takes the output of a multi-box detection model, and converts it into an NDList that contains the object detections.
MultiBoxDetection(MultiBoxDetection.Builder) - Constructor for class ai.djl.modality.cv.MultiBoxDetection
Creates a new instance of MultiBoxDetection with the arguments from the given MultiBoxDetection.Builder.
MultiBoxDetection.Builder - Class in ai.djl.modality.cv
The Builder to construct a MultiBoxDetection object.
MultiBoxPrior - Class in ai.djl.modality.cv
MultiBoxPrior is the class that generates anchor boxes that act as priors for object detection.
MultiBoxPrior(MultiBoxPrior.Builder) - Constructor for class ai.djl.modality.cv.MultiBoxPrior
Creates a new instance of MultiBoxPrior with the arguments from the given MultiBoxPrior.Builder.
MultiBoxPrior.Builder - Class in ai.djl.modality.cv
The Builder to construct a MultiBoxPrior object.
MultiBoxTarget - Class in ai.djl.modality.cv
MultiBoxTarget is the class that computes the training targets for training a Single Shot Detection (SSD) models.
MultiBoxTarget(MultiBoxTarget.Builder) - Constructor for class ai.djl.modality.cv.MultiBoxTarget
Creates a new instance of MultiBoxTarget with the arguments from the given MultiBoxTarget.Builder.
MultiBoxTarget.Builder - Class in ai.djl.modality.cv
The Builder to construct a MultiBoxTarget object.
multiFactor() - Static method in interface ai.djl.training.tracker.Tracker
Returns a new instance of MultiFactorTracker.Builder that can build an MultiFactorTracker.
MultiFactorTracker - Class in ai.djl.training.tracker
MultiFactorTracker is an implementation of Tracker which returns piecewise constant values for fixed numbers of steps.
MultiFactorTracker(MultiFactorTracker.Builder) - Constructor for class ai.djl.training.tracker.MultiFactorTracker
Creates a new instance of MultiFactorTracker.
MultiFactorTracker.Builder - Class in ai.djl.training.tracker
The Builder to construct an MultiFactorTracker object.
MULTIPLE_CHOICE - Static variable in interface ai.djl.Application.NLP
An application to represent a multiple choice question.

N

Nag - Class in ai.djl.training.optimizer
Nag is a Nesterov accelerated gradient optimizer.
Nag(Nag.Builder) - Constructor for class ai.djl.training.optimizer.Nag
Creates a new instance of Nag optimizer.
nag() - Static method in class ai.djl.training.optimizer.Optimizer
Returns a new instance of Nag.Builder that can build an Nag optimizer.
Nag.Builder - Class in ai.djl.training.optimizer
The Builder to construct an Nag object.
name - Variable in class ai.djl.ndarray.BaseNDManager
 
name - Variable in class ai.djl.ndarray.NDArrayAdapter
 
name - Variable in class ai.djl.repository.AbstractRepository
 
name - Variable in class ai.djl.training.hyperparameter.param.Hyperparameter
 
nano() - Method in class ai.djl.nn.transformer.BertBlock.Builder
Tiny config for testing on laptops.
NDArray - Interface in ai.djl.ndarray
An interface representing an n-dimensional array.
NDArrayAdapter - Class in ai.djl.ndarray
A base implementation of the NDArray that does nothing.
NDArrayAdapter(NDManager, NDManager, Shape, DataType, String) - Constructor for class ai.djl.ndarray.NDArrayAdapter
 
NDArrayIndexer - Class in ai.djl.ndarray.index
A helper class for NDArray implementations for operations with an NDIndex.
NDArrayIndexer() - Constructor for class ai.djl.ndarray.index.NDArrayIndexer
 
NDArrays - Class in ai.djl.ndarray
This class contains various methods for manipulating NDArrays.
NDArraySupplier - Interface in ai.djl.translate
Represents a supplier of NDArray.
NDImageUtils - Class in ai.djl.modality.cv.util
NDImageUtils is an image processing utility to load, reshape, and convert images using NDArray images.
NDIndex - Class in ai.djl.ndarray.index
The NDIndex allows you to specify a subset of an NDArray that can be used for fetching or updating.
NDIndex() - Constructor for class ai.djl.ndarray.index.NDIndex
Creates an empty NDIndex to append values to.
NDIndex(String, Object...) - Constructor for class ai.djl.ndarray.index.NDIndex
Creates a NDIndex given the index values.
NDIndex(long...) - Constructor for class ai.djl.ndarray.index.NDIndex
Creates an NDIndex with the given indices as specified values on the NDArray.
NDIndexAll - Class in ai.djl.ndarray.index.dim
An NDIndexElement to return all values in a particular dimension.
NDIndexAll() - Constructor for class ai.djl.ndarray.index.dim.NDIndexAll
 
NDIndexBooleans - Class in ai.djl.ndarray.index.dim
An NDIndexElement to return values based on a mask binary NDArray.
NDIndexBooleans(NDArray) - Constructor for class ai.djl.ndarray.index.dim.NDIndexBooleans
Constructs a NDIndexBooleans instance with specified mask binary NDArray.
NDIndexElement - Interface in ai.djl.ndarray.index.dim
An index for particular dimensions created by NDIndex.
NDIndexFixed - Class in ai.djl.ndarray.index.dim
An NDIndexElement that returns only a specific value in the corresponding dimension.
NDIndexFixed(long) - Constructor for class ai.djl.ndarray.index.dim.NDIndexFixed
Constructs a NDIndexFixed instance with specified dimension.
NDIndexFullPick - Class in ai.djl.ndarray.index.full
A simplified representation of a pick-based NDIndex.
NDIndexFullSlice - Class in ai.djl.ndarray.index.full
An index as a slice on all dimensions where some dimensions can be squeezed.
NDIndexPick - Class in ai.djl.ndarray.index.dim
An NDIndexElement that gets elements by index in the specified axis.
NDIndexPick(NDArray) - Constructor for class ai.djl.ndarray.index.dim.NDIndexPick
Constructs a pick.
NDIndexSlice - Class in ai.djl.ndarray.index.dim
An NDIndexElement that returns a range of values in the specified dimension.
NDIndexSlice(Long, Long, Long) - Constructor for class ai.djl.ndarray.index.dim.NDIndexSlice
Constructs a NDIndexSlice instance with specified range and step.
NDList - Class in ai.djl.ndarray
An NDList represents a sequence of NDArrays with names.
NDList() - Constructor for class ai.djl.ndarray.NDList
Constructs an empty NDList.
NDList(int) - Constructor for class ai.djl.ndarray.NDList
Constructs an empty NDList with the specified initial capacity.
NDList(NDArray...) - Constructor for class ai.djl.ndarray.NDList
Constructs and initiates an NDList with the specified NDArrays.
NDList(Collection<NDArray>) - Constructor for class ai.djl.ndarray.NDList
Constructs and initiates an NDList with the specified NDArrays.
NDManager - Interface in ai.djl.ndarray
NDArray managers are used to create NDArrays (n-dimensional array on native engine).
NDResource - Interface in ai.djl.ndarray
An object which is managed by an NDManager and tracks the manager it is attached to.
NDUtils - Class in ai.djl.ndarray
A class containing utility methods for NDArray operations.
neg() - Method in interface ai.djl.ndarray.NDArray
Returns the numerical negative NDArray element-wise.
neg() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the numerical negative NDArray element-wise.
negi() - Method in interface ai.djl.ndarray.NDArray
Returns the numerical negative NDArray element-wise in place.
negi() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the numerical negative NDArray element-wise in place.
neq(Number) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Not equals" comparison.
neq(NDArray) - Method in interface ai.djl.ndarray.NDArray
Returns the boolean NDArray for element-wise "Not equals" comparison.
neq(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Not equals" comparison.
neq(NDArray) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the boolean NDArray for element-wise "Not equals" comparison.
neq(NDArray, Number) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Not equals" comparison.
neq(Number, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Not equals" comparison.
neq(NDArray, NDArray) - Static method in class ai.djl.ndarray.NDArrays
Returns the boolean NDArray for element-wise "Not equals" comparison.
newBaseManager() - Method in class ai.djl.engine.Engine
Creates a new top-level NDManager.
newBaseManager(Device) - Method in class ai.djl.engine.Engine
Creates a new top-level NDManager with specified Device.
newBaseManager() - Static method in interface ai.djl.ndarray.NDManager
Creates a new top-level NDManager.
newBaseManager(Device) - Static method in interface ai.djl.ndarray.NDManager
Creates a new top-level NDManager with specified Device.
newBaseManager(Device, String) - Static method in interface ai.djl.ndarray.NDManager
Creates a new top-level NDManager with specified Device and engine.
newBlock(Model, Path, Map<String, ?>) - Method in interface ai.djl.nn.BlockFactory
Constructs the uninitialized block.
newGradientCollector() - Method in class ai.djl.engine.Engine
Returns a new instance of GradientCollector.
newGradientCollector() - Method in class ai.djl.training.Trainer
Returns a new instance of GradientCollector.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in class ai.djl.modality.cv.translator.BigGANTranslatorFactory
Returns a new instance of the Translator class.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in class ai.djl.modality.cv.translator.ImageClassificationTranslatorFactory
Returns a new instance of the Translator class.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in class ai.djl.modality.cv.translator.InstanceSegmentationTranslatorFactory
Returns a new instance of the Translator class.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in class ai.djl.modality.cv.translator.SimplePoseTranslatorFactory
Returns a new instance of the Translator class.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in class ai.djl.modality.cv.translator.SingleShotDetectionTranslatorFactory
Returns a new instance of the Translator class.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in class ai.djl.modality.cv.translator.StyleTransferTranslatorFactory
Returns a new instance of the Translator class.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in class ai.djl.modality.cv.translator.YoloTranslatorFactory
Returns a new instance of the Translator class.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in class ai.djl.modality.cv.translator.YoloV5TranslatorFactory
Returns a new instance of the Translator class.
newInstance(String) - Static method in interface ai.djl.Model
Creates an empty model instance.
newInstance(String, Device) - Static method in interface ai.djl.Model
Creates an empty model instance on the specified Device.
newInstance(String, String) - Static method in interface ai.djl.Model
Creates an empty model instance on the specified Device and engine.
newInstance(String, Device, String) - Static method in interface ai.djl.Model
Creates an empty model instance on the specified Device and engine.
newInstance(NDManager) - Static method in interface ai.djl.nn.SymbolBlock
Creates an empty SymbolBlock instance.
newInstance(String, Path) - Static method in interface ai.djl.repository.Repository
Creates a new instance of a repository with a name and url.
newInstance(String, String) - Static method in interface ai.djl.repository.Repository
Creates a new instance of a repository with a name and url.
newInstance(String, URI) - Method in interface ai.djl.repository.RepositoryFactory
Creates a new instance of a repository with a name and url.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in class ai.djl.translate.DefaultTranslatorFactory
Returns a new instance of the Translator class.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in class ai.djl.translate.ServingTranslatorFactory
Returns a new instance of the Translator class.
newInstance(Class<?>, Class<?>, Model, Map<String, ?>) - Method in interface ai.djl.translate.TranslatorFactory
Returns a new instance of the Translator class.
newModel(String, Device) - Method in class ai.djl.engine.Engine
Constructs a new model.
newParameterServer(Optimizer) - Method in class ai.djl.engine.Engine
Returns a new instance of ParameterServer.
newPredictor(Translator<I, O>, Device) - Method in class ai.djl.BaseModel
Creates a new Predictor based on the model.
newPredictor(Translator<I, O>) - Method in interface ai.djl.Model
Creates a new Predictor based on the model on the current device.
newPredictor(Translator<I, O>, Device) - Method in interface ai.djl.Model
Creates a new Predictor based on the model.
newPredictor() - Method in class ai.djl.repository.zoo.ZooModel
Creates a new Predictor based on the model with the default translator.
newPredictor(Device) - Method in class ai.djl.repository.zoo.ZooModel
Creates a new Predictor based on the model with the default translator and a specified device.
newPredictor(Translator<P, Q>, Device) - Method in class ai.djl.repository.zoo.ZooModel
Creates a new Predictor based on the model.
newSubManager() - Method in class ai.djl.ndarray.BaseNDManager
Creates a child NDManager.
newSubManager() - Method in interface ai.djl.ndarray.NDManager
Creates a child NDManager.
newSubManager(Device) - Method in interface ai.djl.ndarray.NDManager
Creates a child NDManager with specified default Device.
newSymbolBlock(NDManager) - Method in class ai.djl.engine.Engine
Construct an empty SymbolBlock for loading.
newTrainer(TrainingConfig) - Method in class ai.djl.BaseModel
Creates a new Trainer instance for a Model.
newTrainer(TrainingConfig) - Method in interface ai.djl.Model
Creates a new Trainer instance for a Model.
newTrainer(TrainingConfig) - Method in class ai.djl.repository.zoo.ZooModel
Creates a new Trainer instance for a Model.
next() - Method in class ai.djl.training.dataset.DataIterable
nextConfig() - Method in interface ai.djl.training.hyperparameter.optimizer.HpOptimizer
Returns the next hyperparameters to test.
nextConfig() - Method in class ai.djl.training.hyperparameter.optimizer.HpORandom
Returns the next hyperparameters to test.
NlpUtils - Class in ai.djl.modality.nlp
Utility functions for processing String and Characters in NLP problems.
nms(List<YoloV5Translator.IntermediateResult>) - Method in class ai.djl.modality.cv.translator.YoloV5Translator
 
NoBatchifyTranslator<I,O> - Interface in ai.djl.translate
A Translator that does not use a Batchifier.
none() - Method in interface ai.djl.ndarray.NDArray
Returns true if none of the elements within this NDArray are non-zero or true.
nonzero() - Method in interface ai.djl.ndarray.NDArray
Returns the indices of elements that are non-zero.
nonzero() - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the indices of elements that are non-zero.
NoopTranslator - Class in ai.djl.translate
A no operational Translator implementation.
NoopTranslator(Batchifier) - Constructor for class ai.djl.translate.NoopTranslator
Constructs a NoopTranslator with the given Batchifier.
NoopTranslator() - Constructor for class ai.djl.translate.NoopTranslator
Constructs a NoopTranslator.
norm() - Method in interface ai.djl.ndarray.NDArray
Returns the norm of this NDArray.
norm(int[]) - Method in interface ai.djl.ndarray.NDArray
Returns the norm of this NDArray.
norm(boolean) - Method in interface ai.djl.ndarray.NDArray
Returns the norm of this NDArray.
norm(int[], boolean) - Method in interface ai.djl.ndarray.NDArray
Returns the norm of this NDArray.
norm(int, int[], boolean) - Method in interface ai.djl.ndarray.NDArray
Returns the norm of this NDArray.
norm(boolean) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the norm of this NDArray.
norm(int, int[], boolean) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns the norm of this NDArray.
NormalInitializer - Class in ai.djl.training.initializer
NormalInitializer initializes weights with random values sampled from a normal distribution with a mean of zero and standard deviation of sigma.
NormalInitializer() - Constructor for class ai.djl.training.initializer.NormalInitializer
Creates an instance of NormalInitializer with a default sigma of 0.01.
NormalInitializer(float) - Constructor for class ai.djl.training.initializer.NormalInitializer
Creates a Normal initializer.
Normalize - Class in ai.djl.modality.cv.transform
A Transform that normalizes an image NDArray of shape CHW or NCHW.
Normalize(float[], float[]) - Constructor for class ai.djl.modality.cv.transform.Normalize
Creates a Normalize Transform that normalizes.
normalize(NDArray, float, float) - Static method in class ai.djl.modality.cv.util.NDImageUtils
Normalizes an image NDArray of shape CHW or NCHW with a single mean and standard deviation to apply to all channels.
normalize(NDArray, float[], float[]) - Static method in class ai.djl.modality.cv.util.NDImageUtils
Normalizes an image NDArray of shape CHW or NCHW with mean and standard deviation.
normalizeDefault(String) - Static method in class ai.djl.modality.nlp.preprocess.UnicodeNormalizer
Normalizes a String using a sensible default normal form.
normalizeHyphens(String) - Static method in class ai.djl.modality.nlp.preprocess.HyphenNormalizer
Replaces hyphen like codepoints by ASCII "-", removes soft hyphens.
notifyListeners(Consumer<TrainingListener>) - Method in class ai.djl.training.Trainer
Executes a method on each of the TrainingListeners.
numChannels() - Method in enum ai.djl.modality.cv.Image.Flag
Returns the number of channels for this flag.
numDimensions() - Method in class ai.djl.nn.convolutional.Conv1d
Returns the number of dimensions of the input.
numDimensions() - Method in class ai.djl.nn.convolutional.Conv1dTranspose
Returns the number of dimensions of the input.
numDimensions() - Method in class ai.djl.nn.convolutional.Conv2d
Returns the number of dimensions of the input.
numDimensions() - Method in class ai.djl.nn.convolutional.Conv2dTranspose
Returns the number of dimensions of the input.
numDimensions() - Method in class ai.djl.nn.convolutional.Conv3d
Returns the number of dimensions of the input.
numDimensions() - Method in class ai.djl.nn.convolutional.Convolution
Returns the number of dimensions of the input.
numDimensions() - Method in class ai.djl.nn.convolutional.Deconvolution
Returns the number of dimensions of the input.
numEmbeddings - Variable in class ai.djl.nn.core.Embedding.BaseBuilder
 
numEmbeddings - Variable in class ai.djl.nn.core.Embedding
 
numEpochs(HpSet) - Method in class ai.djl.training.hyperparameter.EasyHpo
Returns the number of epochs to train for the current hyperparameter set.
numHyperParameterTests() - Method in class ai.djl.training.hyperparameter.EasyHpo
Returns the number of hyperparameter sets to train with.
numLayers - Variable in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
 
numLayers - Variable in class ai.djl.nn.recurrent.RecurrentBlock
 

O

OBJECT_DETECTION - Static variable in interface ai.djl.Application.CV
An application that finds zero or more objects in an image, the object class (see image classification), and their locations as a BoundingBox.
ObjectDetectionBuilder() - Constructor for class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder
 
ObjectDetectionTranslator - Class in ai.djl.modality.cv.translator
A BaseImageTranslator that post-process the NDArray into DetectedObjects with boundaries.
ObjectDetectionTranslator(ObjectDetectionTranslator.ObjectDetectionBuilder<?>) - Constructor for class ai.djl.modality.cv.translator.ObjectDetectionTranslator
Creates the ObjectDetectionTranslator from the given builder.
ObjectDetectionTranslator.ObjectDetectionBuilder<T extends ObjectDetectionTranslator.ObjectDetectionBuilder> - Class in ai.djl.modality.cv.translator
The base builder for the object detection translator.
ObjectDetectionTranslatorFactory - Class in ai.djl.modality.cv.translator
An abstract TranslatorFactory that creates a ObjectDetectionTranslator instance.
ObjectDetectionTranslatorFactory() - Constructor for class ai.djl.modality.cv.translator.ObjectDetectionTranslatorFactory
 
of(String) - Static method in class ai.djl.Application
Converts a path string to a Application.
of(String, int) - Static method in class ai.djl.Device
Returns a Device with device type and device id.
oneHot(int) - Method in interface ai.djl.ndarray.NDArray
Returns a one-hot NDArray.
oneHot(int, DataType) - Method in interface ai.djl.ndarray.NDArray
Returns a one-hot NDArray.
oneHot(int, float, float, DataType) - Method in interface ai.djl.ndarray.NDArray
Returns a one-hot NDArray.
oneHot(int, float, float, DataType) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns a one-hot NDArray.
onEpoch(Trainer) - Method in class ai.djl.training.listener.EpochTrainingListener
Listens to the end of an epoch during training.
onEpoch(Trainer) - Method in class ai.djl.training.listener.EvaluatorTrainingListener
Listens to the end of an epoch during training.
onEpoch(Trainer) - Method in class ai.djl.training.listener.LoggingTrainingListener
Listens to the end of an epoch during training.
onEpoch(Trainer) - Method in class ai.djl.training.listener.SaveModelTrainingListener
Listens to the end of an epoch during training.
onEpoch(Trainer) - Method in class ai.djl.training.listener.TimeMeasureTrainingListener
Listens to the end of an epoch during training.
onEpoch(Trainer) - Method in interface ai.djl.training.listener.TrainingListener
Listens to the end of an epoch during training.
onEpoch(Trainer) - Method in class ai.djl.training.listener.TrainingListenerAdapter
Listens to the end of an epoch during training.
ones(Shape, DataType) - Method in class ai.djl.ndarray.BaseNDManager
Creates an instance of NDArray with specified Shape filled with ones.
ones(Shape, DataType) - Method in interface ai.djl.ndarray.NDManager
Creates an instance of NDArray with specified Shape filled with ones.
ones(Shape) - Method in interface ai.djl.ndarray.NDManager
Creates an instance of NDArray with specified Shape filled with ones.
ones(Shape, DataType, Device) - Method in interface ai.djl.ndarray.NDManager
Creates an instance of NDArray with specified Device, Shape, and DataType filled with ones.
ONES - Static variable in interface ai.djl.training.initializer.Initializer
 
onesLike() - Method in interface ai.djl.ndarray.NDArray
Returns an NDArray of ones with the same Shape, DataType and SparseFormat as the input NDArray.
onTrainingBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.DivergenceCheckTrainingListener
Listens to the end of training one batch of data during training.
onTrainingBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.EvaluatorTrainingListener
Listens to the end of training one batch of data during training.
onTrainingBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.LoggingTrainingListener
Listens to the end of training one batch of data during training.
onTrainingBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.MemoryTrainingListener
Listens to the end of training one batch of data during training.
onTrainingBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.TimeMeasureTrainingListener
Listens to the end of training one batch of data during training.
onTrainingBatch(Trainer, TrainingListener.BatchData) - Method in interface ai.djl.training.listener.TrainingListener
Listens to the end of training one batch of data during training.
onTrainingBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.TrainingListenerAdapter
Listens to the end of training one batch of data during training.
onTrainingBegin(Trainer) - Method in class ai.djl.training.listener.EpochTrainingListener
Listens to the beginning of training.
onTrainingBegin(Trainer) - Method in class ai.djl.training.listener.EvaluatorTrainingListener
Listens to the beginning of training.
onTrainingBegin(Trainer) - Method in class ai.djl.training.listener.LoggingTrainingListener
Listens to the beginning of training.
onTrainingBegin(Trainer) - Method in interface ai.djl.training.listener.TrainingListener
Listens to the beginning of training.
onTrainingBegin(Trainer) - Method in class ai.djl.training.listener.TrainingListenerAdapter
Listens to the beginning of training.
onTrainingEnd(Trainer) - Method in class ai.djl.training.listener.EpochTrainingListener
Listens to the end of training.
onTrainingEnd(Trainer) - Method in class ai.djl.training.listener.LoggingTrainingListener
Listens to the end of training.
onTrainingEnd(Trainer) - Method in class ai.djl.training.listener.MemoryTrainingListener
Listens to the end of training.
onTrainingEnd(Trainer) - Method in class ai.djl.training.listener.SaveModelTrainingListener
Listens to the end of training.
onTrainingEnd(Trainer) - Method in class ai.djl.training.listener.TimeMeasureTrainingListener
Listens to the end of training.
onTrainingEnd(Trainer) - Method in interface ai.djl.training.listener.TrainingListener
Listens to the end of training.
onTrainingEnd(Trainer) - Method in class ai.djl.training.listener.TrainingListenerAdapter
Listens to the end of training.
onValidationBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.EvaluatorTrainingListener
Listens to the end of validating one batch of data during validation.
onValidationBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.LoggingTrainingListener
Listens to the end of validating one batch of data during validation.
onValidationBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.MemoryTrainingListener
Listens to the end of validating one batch of data during validation.
onValidationBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.TimeMeasureTrainingListener
Listens to the end of validating one batch of data during validation.
onValidationBatch(Trainer, TrainingListener.BatchData) - Method in interface ai.djl.training.listener.TrainingListener
Listens to the end of validating one batch of data during validation.
onValidationBatch(Trainer, TrainingListener.BatchData) - Method in class ai.djl.training.listener.TrainingListenerAdapter
Listens to the end of validating one batch of data during validation.
OPENMP - Static variable in class ai.djl.engine.StandardCapabilities
 
openStream(Artifact.Item, String) - Method in class ai.djl.repository.AbstractRepository
Returns an InputStream for an item in a repository.
openStream(Artifact.Item, String) - Method in interface ai.djl.repository.Repository
Returns an InputStream for an item in a repository.
optApplication(Application) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets the model application for this criteria.
optApplySoftmax(boolean) - Method in class ai.djl.modality.cv.translator.ImageClassificationTranslator.Builder
Sets whether to apply softmax when processing output.
optArgument(String, Object) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets the optional model loading argument for this criteria.
optArguments(Map<String, Object>) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets an extra model loading argument for this criteria.
optArray(NDArray) - Method in class ai.djl.nn.Parameter.Builder
Sets the array of the Parameter.
optArtifactId(String) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets optional artifactId of the ModelLoader for this criteria.
optAttentionHeadCount(int) - Method in class ai.djl.nn.transformer.BertBlock.Builder
Sets the number of attention heads to use in each transformer block.
optAttentionProbsDropoutProb(float) - Method in class ai.djl.nn.transformer.ScaledDotProductAttentionBlock.Builder
Sets the probability of applying dropout to the attention probability distribution.
optAxis(int) - Method in class ai.djl.nn.norm.BatchNorm.Builder
Set the axis in which channel is specified.
optBackgroundId(int) - Method in class ai.djl.modality.cv.MultiBoxDetection.Builder
Sets the class ID for the background.
optBatchFirst(boolean) - Method in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
Sets the optional batchFirst flag that indicates whether the input is batch major or not.
optBatchifier(Batchifier) - Method in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
Sets the Batchifier for the Translator.
optBatchifier(Batchifier) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Sets the Batchifier for the Translator.
optBeginNumUpdate(int) - Method in class ai.djl.training.optimizer.Optimizer.OptimizerBuilder
Sets the initial value of the number of updates.
optBeta1(float) - Method in class ai.djl.training.optimizer.Adam.Builder
Sets the decay rate for the first moment estimates.
optBeta2(float) - Method in class ai.djl.training.optimizer.Adam.Builder
Sets the decay rate for the second moment estimates.
optBias(boolean) - Method in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
Sets the optional parameter of whether to include a bias vector.
optBias(boolean) - Method in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
Sets the optional parameter of whether to include a bias vector.
optBias(boolean) - Method in class ai.djl.nn.core.Linear.Builder
Sets the optional parameter that indicates whether to include a bias vector with default value of true.
optBidirectional(boolean) - Method in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
Sets the optional parameter that indicates whether to use bidirectional recurrent layers.
optBlock(Block) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets an optional model Block for this criteria.
optCenter(boolean) - Method in class ai.djl.nn.norm.BatchNorm.Builder
If True, add offset of `beta` to normalized tensor.
optCenter(boolean) - Method in class ai.djl.nn.norm.LayerNorm.Builder
If True, add offset of `beta` to normalized tensor.
optCentered(boolean) - Method in class ai.djl.training.optimizer.RmsProp.Builder
Sets which version of RMSProp to use.
optClip(boolean) - Method in class ai.djl.modality.cv.MultiBoxDetection.Builder
Sets the boolean parameter that indicates whether to clip out-of-boundary boxes.
optClip(boolean) - Method in class ai.djl.modality.cv.MultiBoxPrior.Builder
Sets the boolean parameter that indicates whether to clip out-of-boundary boxes.
optClipGrad(float) - Method in class ai.djl.training.optimizer.Optimizer.OptimizerBuilder
Sets the value of the \(clipGrad\).
optDataBatchifier(Batchifier) - Method in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
Sets the Batchifier for the data.
optDefaultItem(T) - Method in class ai.djl.nn.core.Embedding.BaseBuilder
Sets whether to use a default item's embedding for undefined items.
optDevice(Device) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets the Device for this criteria.
optDevice(Device) - Method in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
Sets the Device.
optDevices(Device[]) - Method in class ai.djl.training.DefaultTrainingConfig
Sets the array of Device available for training.
optDilation(Shape) - Method in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
Sets the dilation along each dimension.
optDilation(Shape) - Method in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
Sets the dilation along each dimension.
optDropRate(float) - Method in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
Sets the drop rate of the dropout on the outputs of each RNN layer, except the last layer.
optEmbeddingSize(int) - Method in class ai.djl.nn.transformer.BertBlock.Builder
Sets the embedding size to use for input tokens.
optEngine(String) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets the engine name for this criteria.
optEpsilon(float) - Method in class ai.djl.nn.norm.BatchNorm.Builder
Sets the epsilon value to prevent division by 0.
optEpsilon(float) - Method in class ai.djl.nn.norm.LayerNorm.Builder
Sets the epsilon value to prevent division by 0.
optEpsilon(float) - Method in class ai.djl.training.optimizer.Adadelta.Builder
Sets \(epsilon\) - a small quantity for numerical stability.
optEpsilon(float) - Method in class ai.djl.training.optimizer.Adagrad.Builder
Sets \(epsilon\) - a small quantity for numerical stability.
optEpsilon(float) - Method in class ai.djl.training.optimizer.Adam.Builder
Sets \(epsilon\) - a small quantity for numerical stability.
optEpsilon(float) - Method in class ai.djl.training.optimizer.RmsProp.Builder
Sets \(epsilon\) - a small quantity for numerical stability.
optExecutorService() - Method in class ai.djl.training.DefaultTrainingConfig
Sets the ExecutorService with the global ForkJoinPool.commonPool().
optExecutorService(ExecutorService) - Method in class ai.djl.training.DefaultTrainingConfig
Sets the ExecutorService to train with multiple threads.
optFactor(float) - Method in class ai.djl.training.tracker.MultiFactorTracker.Builder
Set the value of the multiplicative factor.
optFallthrough(AbstractIndexedEmbedding<T>) - Method in class ai.djl.nn.core.Embedding.BaseBuilder
Sets a custom handler for items not found in the embedding.
optFilter(String, String) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets an extra search filter for this criteria.
optFilters(Map<String, String>) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets the extra search filters for this criteria.
optFinalValue(float) - Method in class ai.djl.training.tracker.CosineTracker.Builder
Sets the final value that the learning rate will remain constant as after the specified max number of updates.
optFlag(Image.Flag) - Method in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
Sets the optional Image.Flag (default is Image.Flag#COLOR).
optForceSuppress(boolean) - Method in class ai.djl.modality.cv.MultiBoxDetection.Builder
Sets the boolean parameter that indicates whether to suppress all detections regardless of class_id.
optGroupId(String) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets optional groupId of the ModelZoo for this criteria.
optGroups(int) - Method in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
Sets the number of group partitions.
optGroups(int) - Method in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
Sets the number of group partitions.
optHasBiases(boolean) - Method in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
Sets the optional biases flag that indicates whether to use biases or not.
optHiddenDropoutProbability(float) - Method in class ai.djl.nn.transformer.BertBlock.Builder
Sets the dropout probabilty in the hidden fully connected networks.
optHiddenSize(int) - Method in class ai.djl.nn.transformer.BertBlock.Builder
Sets the size of the hidden layers in the fully connected networks used.
optIgnoreLabel(float) - Method in class ai.djl.modality.cv.MultiBoxTarget.Builder
Sets the label for ignored anchors.
Optimizer - Class in ai.djl.training.optimizer
An Optimizer updates the weight parameters to minimize the loss function.
Optimizer(Optimizer.OptimizerBuilder<?>) - Constructor for class ai.djl.training.optimizer.Optimizer
Creates a new instance of Optimizer.
Optimizer.OptimizerBuilder<T extends Optimizer.OptimizerBuilder> - Class in ai.djl.training.optimizer
The Builder to construct an Optimizer.
OptimizerBuilder() - Constructor for class ai.djl.training.optimizer.Optimizer.OptimizerBuilder
 
optIncludeTokenTypes(boolean) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Sets the if include token types for the Translator.
optIncludeValidLengths(boolean) - Method in class ai.djl.translate.PaddingStackBatchifier.Builder
Sets whether to include the valid lengths (length of non-padded data) for each array.
optInitializer(Initializer) - Method in class ai.djl.nn.Parameter.Builder
Sets the Initializer of the Parameter.
optInitializer(Initializer, Parameter.Type) - Method in class ai.djl.training.DefaultTrainingConfig
Sets the Initializer to use for the parameters (default from paper).
optInitializer(Initializer, String) - Method in class ai.djl.training.DefaultTrainingConfig
Sets the Initializer to use for the parameters (default from paper).
optInitializer(Initializer, Predicate<Parameter>) - Method in class ai.djl.training.DefaultTrainingConfig
Sets the Initializer to use for the parameters (default from paper).
optIouThreshold(float) - Method in class ai.djl.modality.cv.MultiBoxTarget.Builder
Sets the anchor-GroundTruth overlap threshold to be regarded as a positive match.
optLabelBatchifier(Batchifier) - Method in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
Sets the Batchifier for the labels.
optLabels(NDArray...) - Method in class ai.djl.training.dataset.ArrayDataset.Builder
Sets the labels for the data in the ArrayDataset.
optLearningRateTracker(Tracker) - Method in class ai.djl.training.optimizer.Adagrad.Builder
Sets the Tracker for this optimizer.
optLearningRateTracker(Tracker) - Method in class ai.djl.training.optimizer.Adam.Builder
Sets the Tracker for this optimizer.
optLearningRateTracker(Tracker) - Method in class ai.djl.training.optimizer.RmsProp.Builder
Sets the Tracker for this optimizer.
optLimit(long) - Method in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
Sets this dataset's limit.
optLocale(String) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Sets the name of the locale for the Translator.
optMaxEdge(int) - Method in class ai.djl.modality.cv.translator.InstanceSegmentationTranslator.Builder
Sets the maximum edge length of the rescaled image.
optMaxLabels(int) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Sets the max number of labels for the Translator.
optMaxLength(int) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Sets the max number of tokens for the Translator.
optMaxSequenceLength(int) - Method in class ai.djl.nn.transformer.BertBlock.Builder
Sets the maximum sequence length this model can process.
optMaxTokens(int) - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Sets the optional limit on the size of the vocabulary.
optMaxUpdates(int) - Method in class ai.djl.training.tracker.FactorTracker.Builder
Sets the maximum number of updates after which the value should remain constant.
optMaxUpdates(int) - Method in class ai.djl.training.tracker.LinearTracker.Builder
Sets the maximum number of updates after which the value should remain constant.
optMaxValue(float) - Method in class ai.djl.training.tracker.LinearTracker.Builder
Sets the maximum value for a positive slope.
optMinFrequency(int) - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Sets the optional parameter that specifies the minimum frequency to consider a token to be part of the DefaultVocabulary.
optMinNegativeSamples(int) - Method in class ai.djl.modality.cv.MultiBoxTarget.Builder
Sets the minimum number of negative samples.
optMinValue(float) - Method in class ai.djl.training.tracker.FactorTracker.Builder
Sets the minimum value.
optMinValue(float) - Method in class ai.djl.training.tracker.LinearTracker.Builder
Sets the minimum value for a negative slope.
optModelName(String) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets an optional model name for this criteria.
optModelPath(Path) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets the optional model path of the ModelLoader for this criteria.
optModelUrls(String) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets optional model urls of the ModelLoader for this criteria.
optModelZoo(ModelZoo) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets optional ModelZoo of the ModelLoader for this criteria.
optMomentum(float) - Method in class ai.djl.nn.norm.BatchNorm.Builder
Set the momentum for moving average.
optMomentum(float) - Method in class ai.djl.training.optimizer.RmsProp.Builder
Sets the momentum factor.
optMomentum(float) - Method in class ai.djl.training.optimizer.Sgd.Builder
Sets the momentum for Sgd.
optNegativeMinigRatio(float) - Method in class ai.djl.modality.cv.MultiBoxTarget.Builder
Sets the max negative to positive samples ratio.
optNegativeMiningThreshold(float) - Method in class ai.djl.modality.cv.MultiBoxTarget.Builder
Sets the threshold used for negative mining.
optNmsThreshold(float) - Method in class ai.djl.modality.cv.MultiBoxDetection.Builder
Sets the non-maximum suppression(NMS) threshold.
optNmsThreshold(float) - Method in class ai.djl.modality.cv.translator.YoloV5Translator.Builder
Sets the NMS threshold.
optNmsTopK(int) - Method in class ai.djl.modality.cv.MultiBoxDetection.Builder
Sets the boolean parameter that indicates whether to clip out-of-boundary boxes.
optNumEmbeddings(int) - Method in class ai.djl.nn.core.Embedding.BaseBuilder
Sets the size of the dictionary of embeddings.
optOffsets(List<Float>) - Method in class ai.djl.modality.cv.MultiBoxPrior.Builder
Sets the value of the center-box offsets across \(x\) and \(y\) dimensions.
optOptimizer(Optimizer) - Method in class ai.djl.training.DefaultTrainingConfig
Sets the Optimizer used during training (default Adam).
optOption(String, String) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets the optional model loading option for this criteria.
optOptions(Map<String, String>) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets the model loading options for this criteria.
optOutPadding(Shape) - Method in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
Sets the out_padding along each dimension.
optOutputType(YoloV5Translator.YoloOutputType) - Method in class ai.djl.modality.cv.translator.YoloV5Translator.Builder
Sets the YoloOutputType.
optPadding(boolean) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Sets the if pad the tokens for the Translator.
optPadding(Shape) - Method in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
Sets the padding along each dimension.
optPadding(Shape) - Method in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
Sets the padding along each dimension.
optPipeline(Pipeline) - Method in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
Sets the Pipeline of Transform to be applied on the data.
optPower(float) - Method in class ai.djl.training.tracker.PolynomialDecayTracker.Builder
Sets the power of the polynomial to decay by.
optPrefetchNumber(int) - Method in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
Sets the number of batches to prefetch at once.
optProgress(Progress) - Method in class ai.djl.repository.zoo.Criteria.Builder
Set the optional Progress.
optRange(float, float) - Method in class ai.djl.training.util.MinMaxScaler
Sets desired range of transformed data.
optRate(float) - Method in class ai.djl.nn.norm.Dropout.Builder
Sets the probability or the fraction of the input that gets dropped out during training time.
optRequiresGrad(boolean) - Method in class ai.djl.nn.Parameter.Builder
Sets if the Parameter requires gradient.
optRescaleSize(double, double) - Method in class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder
Sets the optional rescale size.
optReservedTokens(Collection<String>) - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Sets the optional parameter that sets the list of reserved tokens.
optReturnState(boolean) - Method in class ai.djl.nn.recurrent.RecurrentBlock.BaseBuilder
Sets the optional flag that indicates whether to return state or not.
optRho(float) - Method in class ai.djl.training.optimizer.Adadelta.Builder
Sets the rho for Adadelta.
optRho(float) - Method in class ai.djl.training.optimizer.RmsProp.Builder
Sets the decay factor for the moving average over the past squared gradient.
optScale(boolean) - Method in class ai.djl.nn.norm.BatchNorm.Builder
If True, multiply result by `gamma`.
optScale(boolean) - Method in class ai.djl.nn.norm.LayerNorm.Builder
If True, multiply result by `gamma`.
optShape(Shape) - Method in class ai.djl.nn.Parameter.Builder
Sets the shape of the Parameter.
optShortEdge(int) - Method in class ai.djl.modality.cv.translator.InstanceSegmentationTranslator.Builder
Sets the shorter edge length of the rescaled image.
optSlope(float) - Method in class ai.djl.training.tracker.LinearTracker.Builder
Sets the value of the linear slope.
optSparseFormat(SparseFormat) - Method in class ai.djl.nn.core.Embedding.BaseBuilder
Sets the optional parameter whether to compute row sparse gradient in the backward calculation.
optSteps(List<Float>) - Method in class ai.djl.modality.cv.MultiBoxPrior.Builder
Sets the step across \(x\) and \(y\) dimensions.
optStride(Shape) - Method in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
Sets the stride of the convolution.
optStride(Shape) - Method in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
Sets the stride of the deconvolution.
optSynset(List<String>) - Method in class ai.djl.modality.cv.translator.BaseImageTranslator.ClassificationBuilder
Sets the potential classes for an image.
optSynsetArtifactName(String) - Method in class ai.djl.modality.cv.translator.BaseImageTranslator.ClassificationBuilder
Sets the name of the synset file listing the potential classes for an image.
optSynsetUrl(String) - Method in class ai.djl.modality.cv.translator.BaseImageTranslator.ClassificationBuilder
Sets the URL of the synset file.
optTargetPipeline(Pipeline) - Method in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
Sets the Pipeline of Transform to be applied on the labels.
optThreshold(float) - Method in class ai.djl.modality.cv.MultiBoxDetection.Builder
Sets the threshold score for a detection to be a positive prediction.
optThreshold(float) - Method in class ai.djl.modality.cv.translator.InstanceSegmentationTranslator.Builder
Sets the threshold for prediction accuracy.
optThreshold(float) - Method in class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder
Sets the threshold for prediction accuracy.
optThreshold(float) - Method in class ai.djl.modality.cv.translator.SimplePoseTranslator.Builder
Sets the threshold for prediction accuracy.
optTokenizer(String) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Sets the name of the tokenizer for the Translator.
optToLowerCase(boolean) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Sets the if convert text to lower case for the Translator.
optTransformerBlockCount(int) - Method in class ai.djl.nn.transformer.BertBlock.Builder
Sets the number of transformer blocks to use.
optTranslator(Translator<I, O>) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets the optional Translator to override default Translator.
optTranslatorFactory(TranslatorFactory) - Method in class ai.djl.repository.zoo.Criteria.Builder
Sets the optional TranslatorFactory to override default Translator.
optTruncation(boolean) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Sets the if truncate the tokens for the Translator.
optTypeDictionarySize(int) - Method in class ai.djl.nn.transformer.BertBlock.Builder
Sets the number of possible token types.
optUnknownToken() - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Sets the optional parameter that specifies the unknown token's string value with ">unk<".
optUnknownToken(String) - Method in class ai.djl.modality.nlp.DefaultVocabulary.Builder
Sets the optional parameter that specifies the unknown token's string value.
optUnknownToken(String) - Method in class ai.djl.modality.nlp.embedding.TrainableWordEmbedding.Builder
Sets the optional String value for the unknown token.
optUseDefault(boolean) - Method in class ai.djl.nn.core.Embedding.BaseBuilder
Sets whether to use a default embedding for undefined items (default true).
optVocab(String) - Method in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder
Sets the name of the vocabulary file for the Translator.
optWarmUpBeginValue(float) - Method in class ai.djl.training.tracker.WarmUpTracker.Builder
Sets the value at the beginning of warm-up mode.
optWarmUpMode(WarmUpTracker.Mode) - Method in class ai.djl.training.tracker.WarmUpTracker.Builder
optWarmUpSteps(int) - Method in class ai.djl.training.tracker.WarmUpTracker.Builder
Sets the number of steps until the point the value is updated in warm-up mode.
optWeightDecays(float) - Method in class ai.djl.training.optimizer.Optimizer.OptimizerBuilder
Sets the value of weight decay.
outPadding - Variable in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
 
outPadding - Variable in class ai.djl.nn.convolutional.Deconvolution
 
Output - Class in ai.djl.modality
A class stores the generic inference results.
Output() - Constructor for class ai.djl.modality.Output
Constructs a Output instance.
Output(int, String) - Constructor for class ai.djl.modality.Output
Constructs a Output with specified requestId, code and message.
overlap(double, double, double, double) - Method in class ai.djl.modality.cv.translator.YoloV5Translator
 

P

pad(List<E>, E, int) - Method in class ai.djl.modality.nlp.bert.BertTokenizer
Pads the tokens to the required length.
padding - Variable in class ai.djl.modality.nlp.translator.QATranslator
 
padding - Variable in class ai.djl.nn.convolutional.Convolution.ConvolutionBuilder
 
padding - Variable in class ai.djl.nn.convolutional.Convolution
 
padding - Variable in class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder
 
padding - Variable in class ai.djl.nn.convolutional.Deconvolution
 
PaddingStackBatchifier - Class in ai.djl.translate
The padding stack batchifier is a StackBatchifier that also pads elements to reach the same length.
PaddingStackBatchifier.Builder - Class in ai.djl.translate
Builder to build a PaddingStackBatchifier.
ParallelBlock - Class in ai.djl.nn
ParallelBlock is a Block whose children form a parallel branch in the network and are combined to produce a single output.
ParallelBlock(Function<List<NDList>, NDList>) - Constructor for class ai.djl.nn.ParallelBlock
Creates a parallel block whose branches are combined to form a single output by the given function.
ParallelBlock(Function<List<NDList>, NDList>, List<Block>) - Constructor for class ai.djl.nn.ParallelBlock
Creates a parallel block whose branches are formed by each block in the list of blocks, and are combined to form a single output by the given function.
Parameter - Class in ai.djl.nn
Parameter is a container class that holds a learnable parameter of a model.
Parameter.Builder - Class in ai.djl.nn
A Builder to construct a Parameter.
Parameter.Type - Enum in ai.djl.nn
Enumerates the types of Parameter.
ParameterList - Class in ai.djl.nn
Represents a set of names and Parameters.
ParameterList() - Constructor for class ai.djl.nn.ParameterList
Create an empty ParameterList.
ParameterList(int) - Constructor for class ai.djl.nn.ParameterList
Constructs an empty ParameterList with the specified initial capacity.
ParameterList(List<String>, List<Parameter>) - Constructor for class ai.djl.nn.ParameterList
Constructs a ParameterList containing the elements of the specified keys and values.
ParameterList(List<Pair<String, Parameter>>) - Constructor for class ai.djl.nn.ParameterList
Constructs a ParameterList containing the elements of the specified list of Pairs.
ParameterList(Map<String, Parameter>) - Constructor for class ai.djl.nn.ParameterList
Constructs a ParameterList containing the elements of the specified map.
parameters - Variable in class ai.djl.nn.AbstractBlock
All direct parameters of this Block.
ParameterServer - Interface in ai.djl.training
An interface for a key-value store to store parameters, and their corresponding gradients.
parameterStore - Variable in class ai.djl.inference.Predictor
 
ParameterStore - Class in ai.djl.training
The ParameterStore contains a map from a parameter to the mirrors of it on other devices.
ParameterStore() - Constructor for class ai.djl.training.ParameterStore
Constructs a new ParameterStore instance.
ParameterStore(NDManager, boolean) - Constructor for class ai.djl.training.ParameterStore
Constructs an empty ParameterStore.
paramPathResolver(String, Map<String, ?>) - Method in class ai.djl.BaseModel
 
parent - Variable in class ai.djl.ndarray.BaseNDManager
 
parse(String) - Static method in class ai.djl.repository.VersionRange
Creates a new version range from a string version range.
percentile(String, int) - Method in class ai.djl.metric.Metrics
Returns a percentile Metric object for the specified metric name.
percentile(Number) - Method in interface ai.djl.ndarray.NDArray
Returns percentile for this NDArray.
percentile(Number, int[]) - Method in interface ai.djl.ndarray.NDArray
Returns median along given dimension(s).
percentile(Number) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns percentile for this NDArray.
percentile(Number, int[]) - Method in class ai.djl.ndarray.NDArrayAdapter
Returns median along given dimension(s).
pipeline - Variable in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
 
pipeline - Variable in class ai.djl.modality.cv.translator.BaseImageTranslator
 
pipeline - Variable in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
 
pipeline - Variable in class ai.djl.training.dataset.RandomAccessDataset
 
Pipeline - Class in ai.djl.translate
Pipeline allows applying multiple transforms on an input NDList.
Pipeline() - Constructor for class ai.djl.translate.Pipeline
Creates a new instance of Pipeline that has no Transform defined yet.
Pipeline(Transform...) - Constructor for class ai.djl.translate.Pipeline
Creates a new instance of Pipeline that can apply the given transforms on its input.
Point - Class in ai.djl.modality.cv.output
A point representing a location in (x,y) coordinate space, specified in double precision.
Point(double, double) - Constructor for class ai.djl.modality.cv.output.Point
Constructs and initializes a point at the specified (x,y) location in the coordinate space.
PointwiseFeedForwardBlock - Class in ai.djl.nn.transformer
Fully connected Feed-Forward network, only applied to the last dimension of the input.
PointwiseFeedForwardBlock(List<Integer>, int, Function<NDList, NDList>) - Constructor for class ai.djl.nn.transformer.PointwiseFeedForwardBlock
Creates a pointwise feed-forward block.
PolynomialDecayTracker - Class in ai.djl.training.tracker
Polynomial decay Tracker.