Index

A B C D E F G H I L M N O P Q R S T U W X Z 
All Classes and Interfaces|All Packages|Constant Field Values|Serialized Form

A

abs() - Method in class ai.djl.engine.rust.RsNDArray
abs(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
acos() - Method in class ai.djl.engine.rust.RsNDArray
acos(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
acosh() - Method in class ai.djl.engine.rust.RsNDArray
adadeltaUpdate(NDList, NDList, float, float, float, float, float) - Method in class ai.djl.engine.rust.RsNDArrayEx
adagradUpdate(NDList, NDList, float, float, float, float, float) - Method in class ai.djl.engine.rust.RsNDArrayEx
adamUpdate(NDList, NDList, float, float, float, float, float, float, float, float, boolean, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
adaptiveAvgPool(long, long[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
adaptiveMaxPool(long, long[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
add(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
add(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
add(Number) - Method in class ai.djl.engine.rust.RsNDArray
addi(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
addi(Number) - Method in class ai.djl.engine.rust.RsNDArray
ai.djl.engine.rust - package ai.djl.engine.rust
Contains classes to interface with the underlying native engine.
ai.djl.engine.rust.zoo - package ai.djl.engine.rust.zoo
Contains the built-in RsModelZoo.
ai.djl.huggingface.tokenizers - package ai.djl.huggingface.tokenizers
Contains classes to interface with the underlying Huggingface tokenizers.
ai.djl.huggingface.tokenizers.jni - package ai.djl.huggingface.tokenizers.jni
Contains classes to interface with the underlying Huggingface tokenizers.
ai.djl.huggingface.translator - package ai.djl.huggingface.translator
Contains Translator classes that leverage Huggingface tokenizers.
ai.djl.huggingface.zoo - package ai.djl.huggingface.zoo
Contains the built-in HfModelZoo.
all() - Method in class ai.djl.engine.rust.RsNDArray
allocateDirect(int) - Method in class ai.djl.engine.rust.RsNDManager
any() - Method in class ai.djl.engine.rust.RsNDArray
arange(float, float, float, int, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
arange(float, float, float, DataType) - Method in class ai.djl.engine.rust.RsNDManager
arange(int, int, int, DataType) - Method in class ai.djl.engine.rust.RsNDManager
argMax() - Method in class ai.djl.engine.rust.RsNDArray
argMax(int) - Method in class ai.djl.engine.rust.RsNDArray
argMax(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
argMaxWithAxis(long, int, boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
argMin() - Method in class ai.djl.engine.rust.RsNDArray
argMin(int) - Method in class ai.djl.engine.rust.RsNDArray
argMin(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
argMinWithAxis(long, int, boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
argSort(int, boolean) - Method in class ai.djl.engine.rust.RsNDArray
argSort(long, int, boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
asin() - Method in class ai.djl.engine.rust.RsNDArray
asin(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
asinh() - Method in class ai.djl.engine.rust.RsNDArray
atan() - Method in class ai.djl.engine.rust.RsNDArray
atan(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
atan2(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
atan2(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
atanh() - Method in class ai.djl.engine.rust.RsNDArray
attach(NDManager) - Method in class ai.djl.engine.rust.RsNDArray
avgPool(Shape, Shape, Shape, boolean, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
avgPool2d(long, long[], long[]) - Static method in class ai.djl.engine.rust.RustLibrary
 

B

batchDecode(long[][]) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the decoded Strings from the input batch ids.
batchDecode(long[][], boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the decoded Strings from the input batch ids.
batchDecode(long, long[][], boolean) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
batchDot(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
batchEncode(long, String[], boolean) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
batchEncode(PairList<String, String>) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input text pair in batch.
batchEncode(PairList<String, String>, boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input text pair in batch.
batchEncode(String[]) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentence in batch.
batchEncode(String[], boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentence in batch.
batchEncode(List<String>) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentence in batch.
batchEncode(List<String>, boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentence in batch.
batchEncodePair(long, String[], String[], boolean) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
batchMatMul(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
batchMatMul(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
batchNorm(NDArray, NDArray, NDArray, NDArray, NDArray, int, float, float, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
batchProcessInput(TranslatorContext, List<QAInput>) - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator
batchProcessInput(TranslatorContext, List<StringPair>) - Method in class ai.djl.huggingface.translator.CrossEncoderTranslator
batchProcessInput(TranslatorContext, List<String>) - Method in class ai.djl.huggingface.translator.FillMaskTranslator
batchProcessInput(TranslatorContext, List<String>) - Method in class ai.djl.huggingface.translator.TextClassificationTranslator
batchProcessInput(TranslatorContext, List<String>) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator
batchProcessInput(TranslatorContext, List<String>) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator
batchProcessOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.CrossEncoderTranslator
batchProcessOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.FillMaskTranslator
batchProcessOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator
batchProcessOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.TextClassificationTranslator
batchProcessOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator
batchProcessOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator
booleanMask(long, long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
booleanMask(NDArray, int) - Method in class ai.djl.engine.rust.RsNDArray
broadcast(long, long[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
broadcast(Shape) - Method in class ai.djl.engine.rust.RsNDArray
build() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Builds the translator.
build() - Method in class ai.djl.huggingface.translator.CrossEncoderTranslator.Builder
Builds the translator.
build() - Method in class ai.djl.huggingface.translator.FillMaskTranslator.Builder
Builds the translator.
build() - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator.Builder
Builds the translator.
build() - Method in class ai.djl.huggingface.translator.TextClassificationTranslator.Builder
Builds the translator.
build() - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator.Builder
Builds the translator.
build() - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator.Builder
Builds the translator.
builder() - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Creates a builder to build a HuggingFaceTokenizer.
builder(HuggingFaceTokenizer) - Static method in class ai.djl.huggingface.translator.CrossEncoderTranslator
Creates a builder to build a CrossEncoderTranslator.
builder(HuggingFaceTokenizer) - Static method in class ai.djl.huggingface.translator.FillMaskTranslator
Creates a builder to build a FillMaskTranslator.
builder(HuggingFaceTokenizer) - Static method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator
Creates a builder to build a QuestionAnsweringTranslator.
builder(HuggingFaceTokenizer) - Static method in class ai.djl.huggingface.translator.TextClassificationTranslator
Creates a builder to build a TextClassificationTranslator.
builder(HuggingFaceTokenizer) - Static method in class ai.djl.huggingface.translator.TextEmbeddingTranslator
Creates a builder to build a TextEmbeddingTranslator.
builder(HuggingFaceTokenizer) - Static method in class ai.djl.huggingface.translator.TokenClassificationTranslator
Creates a builder to build a TokenClassificationTranslator.
builder(HuggingFaceTokenizer, Map<String, ?>) - Static method in class ai.djl.huggingface.translator.CrossEncoderTranslator
Creates a builder to build a CrossEncoderTranslator.
builder(HuggingFaceTokenizer, Map<String, ?>) - Static method in class ai.djl.huggingface.translator.FillMaskTranslator
Creates a builder to build a FillMaskTranslator.
builder(HuggingFaceTokenizer, Map<String, ?>) - Static method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator
Creates a builder to build a QuestionAnsweringTranslator.
builder(HuggingFaceTokenizer, Map<String, ?>) - Static method in class ai.djl.huggingface.translator.TextClassificationTranslator
Creates a builder to build a TextClassificationTranslator.
builder(HuggingFaceTokenizer, Map<String, ?>) - Static method in class ai.djl.huggingface.translator.TextEmbeddingTranslator
Creates a builder to build a TextEmbeddingTranslator.
builder(HuggingFaceTokenizer, Map<String, ?>) - Static method in class ai.djl.huggingface.translator.TokenClassificationTranslator
Creates a builder to build a TokenClassificationTranslator.
builder(Map<String, ?>) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Creates a builder to build a HuggingFaceTokenizer.
buildSentence(List<String>) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer

C

cbrt() - Method in class ai.djl.engine.rust.RsNDArray
ceil() - Method in class ai.djl.engine.rust.RsNDArray
ceil(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
CharSpan - Class in ai.djl.huggingface.tokenizers.jni
A class holds character span information.
CharSpan(int, int) - Constructor for class ai.djl.huggingface.tokenizers.jni.CharSpan
Constructs a new CharSpan instance.
checkStatus() - Static method in class ai.djl.huggingface.tokenizers.jni.LibUtils
 
clip(long, double, double) - Static method in class ai.djl.engine.rust.RustLibrary
 
clip(Number, Number) - Method in class ai.djl.engine.rust.RsNDArray
close() - Method in class ai.djl.engine.rust.RsModel
close() - Method in class ai.djl.engine.rust.RsNDArray
close() - Method in class ai.djl.engine.rust.RsSymbolBlock
close() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
complex() - Method in class ai.djl.engine.rust.RsNDArray
complex(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
concat(long[], int) - Static method in class ai.djl.engine.rust.RustLibrary
 
concat(NDList, int) - Method in class ai.djl.engine.rust.RsNDArrayEx
configure(Map<String, ?>) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Configures the builder with the arguments.
configure(Map<String, ?>) - Method in class ai.djl.huggingface.translator.CrossEncoderTranslator.Builder
Configures the builder with the model arguments.
configure(Map<String, ?>) - Method in class ai.djl.huggingface.translator.FillMaskTranslator.Builder
Configures the builder with the model arguments.
configure(Map<String, ?>) - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator.Builder
Configures the builder with the model arguments.
configure(Map<String, ?>) - Method in class ai.djl.huggingface.translator.TextClassificationTranslator.Builder
Configures the builder with the model arguments.
configure(Map<String, ?>) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator.Builder
Configures the builder with the model arguments.
configure(Map<String, ?>) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator.Builder
Configures the builder with the model arguments.
contentEqual(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
contentEquals(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
contentEquals(Number) - Method in class ai.djl.engine.rust.RsNDArray
convolution(NDArray, NDArray, NDArray, Shape, Shape, Shape, int) - Method in class ai.djl.engine.rust.RsNDArrayEx
cos() - Method in class ai.djl.engine.rust.RsNDArray
cos(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
cosh() - Method in class ai.djl.engine.rust.RsNDArray
cosh(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
countNonzero() - Method in class ai.djl.engine.rust.RsNDArray
countNonzero(int) - Method in class ai.djl.engine.rust.RsNDArray
countNonzero(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
countNonzeroWithAxis(long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
create(Shape, DataType) - Method in class ai.djl.engine.rust.RsNDManager
create(String[], Charset, Shape) - Method in class ai.djl.engine.rust.RsNDManager
create(Buffer, Shape, DataType) - Method in class ai.djl.engine.rust.RsNDManager
createBpeTokenizer(String, String) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
createCoo(Buffer, long[][], Shape) - Method in class ai.djl.engine.rust.RsNDManager
createTokenizer(String, String) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
createTokenizerFromString(String) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
CrossEncoderTranslator - Class in ai.djl.huggingface.translator
The translator for Huggingface cross encoder model.
CrossEncoderTranslator.Builder - Class in ai.djl.huggingface.translator
The builder for question answering translator.
cumProd(int) - Method in class ai.djl.engine.rust.RsNDArray
cumProd(int, DataType) - Method in class ai.djl.engine.rust.RsNDArray
cumProd(long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
cumProdWithAxis(long, int, boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
cumProdWithType(long, int, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
cumSum() - Method in class ai.djl.engine.rust.RsNDArray
cumSum(int) - Method in class ai.djl.engine.rust.RsNDArray
cumSum(long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 

D

decode(long[]) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the decoded String from the input ids.
decode(long[], boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the decoded String from the input ids.
decode(long, long[], boolean) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
deconvolution(NDArray, NDArray, NDArray, Shape, Shape, Shape, Shape, int) - Method in class ai.djl.engine.rust.RsNDArrayEx
deleteEncoding(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
deleteModel(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
deleteTensor(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
deleteTokenizer(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
detach() - Method in class ai.djl.engine.rust.RsNDArray
disablePadding(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
disableTruncation(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
div(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
div(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
div(Number) - Method in class ai.djl.engine.rust.RsNDArray
divi(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
divi(Number) - Method in class ai.djl.engine.rust.RsNDArray
dot(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
dot(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
dropout(NDArray, float, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
duplicate() - Method in class ai.djl.engine.rust.RsNDArray
duplicate(long) - Static method in class ai.djl.engine.rust.RustLibrary
 

E

elu(float) - Method in class ai.djl.engine.rust.RsNDArrayEx
elu(long, float) - Static method in class ai.djl.engine.rust.RustLibrary
 
embedding(NDArray, NDArray, SparseFormat) - Method in class ai.djl.engine.rust.RsNDArrayEx
encode(long, String, boolean) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
encode(String) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentence.
encode(String[]) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentences.
encode(String[], boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentences.
encode(String, boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentence.
encode(String, String) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentence.
encode(String, String, boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentence.
encode(List<String>) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentences.
encode(List<String>, boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the Encoding of the input sentences.
encodeDual(long, String, String, boolean) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
encodeList(long, String[], boolean) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
Encoding - Class in ai.djl.huggingface.tokenizers
A class holds token encoding information.
Encoding(long[], long[], String[], long[], long[], long[], CharSpan[], boolean, Encoding[]) - Constructor for class ai.djl.huggingface.tokenizers.Encoding
 
ENGINE_NAME - Static variable in class ai.djl.engine.rust.RsEngine
 
eq(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
eq(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
eq(Number) - Method in class ai.djl.engine.rust.RsNDArray
equals(Object) - Method in class ai.djl.engine.rust.RsNDArray
erf() - Method in class ai.djl.engine.rust.RsNDArray
erf(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
erfinv() - Method in class ai.djl.engine.rust.RsNDArray
erfinv(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
exceedMaxLength() - Method in class ai.djl.huggingface.tokenizers.Encoding
Returns if tokens exceed max length.
exp() - Method in class ai.djl.engine.rust.RsNDArray
exp(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
expandDims(int) - Method in class ai.djl.engine.rust.RsNDArray
expandDims(long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
eye(int, int, int, DataType) - Method in class ai.djl.engine.rust.RsNDManager
eye(int, int, int, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 

F

fft(long, long) - Method in class ai.djl.engine.rust.RsNDArray
fft2(long[], long[]) - Method in class ai.djl.engine.rust.RsNDArray
FillMaskTranslator - Class in ai.djl.huggingface.translator
The translator for Huggingface fill mask model.
FillMaskTranslator.Builder - Class in ai.djl.huggingface.translator
The builder for fill mask translator.
FillMaskTranslatorFactory - Class in ai.djl.huggingface.translator
A TranslatorFactory that creates a FillMaskTranslator instance.
FillMaskTranslatorFactory() - Constructor for class ai.djl.huggingface.translator.FillMaskTranslatorFactory
 
finalize() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
flatten() - Method in class ai.djl.engine.rust.RsNDArray
flatten(int, int) - Method in class ai.djl.engine.rust.RsNDArray
flatten(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
flattenWithDims(long, int, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
flip(int...) - Method in class ai.djl.engine.rust.RsNDArray
flip(long, int[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
floor() - Method in class ai.djl.engine.rust.RsNDArray
floor(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.engine.rust.RsSymbolBlock
from(NDArray) - Method in class ai.djl.engine.rust.RsNDManager
full(float, long[], int, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
full(Shape, float, DataType) - Method in class ai.djl.engine.rust.RsNDManager
fullSlice(long, long[], long[], long[]) - Static method in class ai.djl.engine.rust.RustLibrary
 

G

gammaln() - Method in class ai.djl.engine.rust.RsNDArray
gather(long, long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
gather(NDArray, int) - Method in class ai.djl.engine.rust.RsNDArray
gatherNd(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
gelu() - Method in class ai.djl.engine.rust.RsNDArrayEx
gelu(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
get(NDArray, NDIndexFullPick) - Method in class ai.djl.engine.rust.RsNDArrayIndexer
get(NDArray, NDIndexFullSlice) - Method in class ai.djl.engine.rust.RsNDArrayIndexer
get(NDArray, NDIndexFullTake) - Method in class ai.djl.engine.rust.RsNDArrayIndexer
getAlternativeEngine() - Method in class ai.djl.engine.rust.RsEngine
getArray() - Method in class ai.djl.engine.rust.RsNDArrayEx
getAttentionMask() - Method in class ai.djl.huggingface.tokenizers.Encoding
Returns the attention masks.
getAttentionMask(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getBatchifier() - Method in class ai.djl.huggingface.translator.CrossEncoderTranslator
getBatchifier() - Method in class ai.djl.huggingface.translator.FillMaskTranslator
getBatchifier() - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator
getBatchifier() - Method in class ai.djl.huggingface.translator.TextClassificationTranslator
getBatchifier() - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator
getBatchifier() - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator
getByteBuffer(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
getCharTokenSpans() - Method in class ai.djl.huggingface.tokenizers.Encoding
Returns char token spans.
getDataType() - Method in class ai.djl.engine.rust.RsNDArray
getDataType(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
getDevice() - Method in class ai.djl.engine.rust.RsNDArray
getDevice(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
getDirectParameters() - Method in class ai.djl.engine.rust.RsSymbolBlock
getEnd() - Method in class ai.djl.huggingface.tokenizers.jni.CharSpan
Returns the end position.
getEngine() - Method in class ai.djl.engine.rust.RsEngineProvider
getEngine() - Method in class ai.djl.engine.rust.RsNDManager
getEngineName() - Method in class ai.djl.engine.rust.RsEngine
getEngineName() - Method in class ai.djl.engine.rust.RsEngineProvider
getEngineRank() - Method in class ai.djl.engine.rust.RsEngineProvider
getGradient() - Method in class ai.djl.engine.rust.RsNDArray
getGroupId() - Method in class ai.djl.engine.rust.zoo.RsModelZoo
getGroupId() - Method in class ai.djl.huggingface.zoo.HfModelZoo
getHandle() - Method in class ai.djl.engine.rust.RsSymbolBlock
Gets the native Rust pointer.
getIds() - Method in class ai.djl.huggingface.tokenizers.Encoding
Returns the token ids.
getIndexer(NDManager) - Method in class ai.djl.engine.rust.RsNDArrayEx
getInputNames(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
getManager() - Method in class ai.djl.engine.rust.RsNDArray
getMaxLength() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the max token length.
getMaxLength(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getModelLoader(String) - Method in class ai.djl.engine.rust.zoo.RsModelZoo
getModelLoader(String) - Method in class ai.djl.huggingface.zoo.HfModelZoo
getModelLoaders() - Method in class ai.djl.engine.rust.zoo.RsModelZoo
getModelLoaders() - Method in class ai.djl.huggingface.zoo.HfModelZoo
getModelZoo() - Method in class ai.djl.engine.rust.zoo.RsZooProvider
getModelZoo() - Method in class ai.djl.huggingface.zoo.HfZooProvider
getName() - Method in class ai.djl.engine.rust.RsNDArray
getNDArrayInternal() - Method in class ai.djl.engine.rust.RsNDArray
getOverflowCount(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getOverflowing() - Method in class ai.djl.huggingface.tokenizers.Encoding
Returns an array of overflowing encodings.
getOverflowing(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getPadding() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the padding policy.
getPaddingStrategy(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getPadToMultipleOf() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the padToMultipleOf for padding.
getPadToMultipleOf(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getRank() - Method in class ai.djl.engine.rust.RsEngine
getShape() - Method in class ai.djl.engine.rust.RsNDArray
getShape(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
getSparseFormat() - Method in class ai.djl.engine.rust.RsNDArray
getSpecialTokenMask() - Method in class ai.djl.huggingface.tokenizers.Encoding
Returns the special token masks.
getSpecialTokenMask(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getStart() - Method in class ai.djl.huggingface.tokenizers.jni.CharSpan
Returns the start position.
getStride() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the stride to use in overflow overlap when truncating sequences longer than the model supports.
getStride(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getSupportedEngines() - Method in class ai.djl.engine.rust.zoo.RsModelZoo
getSupportedEngines() - Method in class ai.djl.huggingface.zoo.HfModelZoo
getSupportedTypes() - Method in class ai.djl.huggingface.translator.FillMaskTranslatorFactory
getSupportedTypes() - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslatorFactory
getSupportedTypes() - Method in class ai.djl.huggingface.translator.TextClassificationTranslatorFactory
getSupportedTypes() - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslatorFactory
getSupportedTypes() - Method in class ai.djl.huggingface.translator.TokenClassificationTranslatorFactory
getTokenCharSpans(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getTokenIds(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getTokens() - Method in class ai.djl.huggingface.tokenizers.Encoding
Returns the tokens.
getTokens(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getTruncation() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the truncation policy.
getTruncationStrategy(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getTypeIds() - Method in class ai.djl.huggingface.tokenizers.Encoding
Returns the token type ids.
getTypeIds(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
getVersion() - Method in class ai.djl.engine.rust.RsEngine
getVersion() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Returns the version of the Huggingface tokenizer.
getWordIds() - Method in class ai.djl.huggingface.tokenizers.Encoding
Returns the word ids.
getWordIds(long) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
globalAvgPool() - Method in class ai.djl.engine.rust.RsNDArrayEx
globalLpPool(float) - Method in class ai.djl.engine.rust.RsNDArrayEx
globalMaxPool() - Method in class ai.djl.engine.rust.RsNDArrayEx
gru(NDArray, NDArray, NDList, boolean, int, double, boolean, boolean, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
gt(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
gt(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
gt(Number) - Method in class ai.djl.engine.rust.RsNDArray
gte(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
gte(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
gte(Number) - Method in class ai.djl.engine.rust.RsNDArray

H

hanningWindow(long) - Method in class ai.djl.engine.rust.RsNDManager
hannWindow(long, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
hasCapability(String) - Method in class ai.djl.engine.rust.RsEngine
hasGradient() - Method in class ai.djl.engine.rust.RsNDArray
hashCode() - Method in class ai.djl.engine.rust.RsNDArray
HfModelZoo - Class in ai.djl.huggingface.zoo
HfModelZoo is a repository that contains HuggingFace models.
HfZooProvider - Class in ai.djl.huggingface.zoo
An Huggingface model zoo provider implements the ZooProvider interface.
HfZooProvider() - Constructor for class ai.djl.huggingface.zoo.HfZooProvider
 
HuggingFaceTokenizer - Class in ai.djl.huggingface.tokenizers
HuggingFaceTokenizer is a Huggingface tokenizer implementation of the Tokenizer interface that converts sentences into token.
HuggingFaceTokenizer.Builder - Class in ai.djl.huggingface.tokenizers
The builder for creating huggingface tokenizer.

I

ifft2(long[], long[]) - Method in class ai.djl.engine.rust.RsNDArray
intern(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
inverse() - Method in class ai.djl.engine.rust.RsNDArray
inverse(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
isCudaAvailable() - Static method in class ai.djl.engine.rust.RustLibrary
 
isInf(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
isInfinite() - Method in class ai.djl.engine.rust.RsNDArray
isNaN() - Method in class ai.djl.engine.rust.RsNDArray
isNaN(long) - Static method in class ai.djl.engine.rust.RustLibrary
 

L

layerNorm(NDArray, Shape, NDArray, NDArray, float) - Method in class ai.djl.engine.rust.RsNDArrayEx
leakyRelu(float) - Method in class ai.djl.engine.rust.RsNDArrayEx
leakyRelu(long, float) - Static method in class ai.djl.engine.rust.RustLibrary
 
LIB - Static variable in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
LibUtils - Class in ai.djl.huggingface.tokenizers.jni
Utilities for finding the Huggingface tokenizer native binary on the System.
linear(NDArray, NDArray, NDArray) - Method in class ai.djl.engine.rust.RsNDArrayEx
linspace(float, float, int, boolean) - Method in class ai.djl.engine.rust.RsNDManager
linspace(float, float, int, int, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
load(Path, String, Map<String, ?>) - Method in class ai.djl.engine.rust.RsModel
loadModel(String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
log() - Method in class ai.djl.engine.rust.RsNDArray
log(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
log10() - Method in class ai.djl.engine.rust.RsNDArray
log10(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
log2() - Method in class ai.djl.engine.rust.RsNDArray
log2(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
logicalAnd(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
logicalAnd(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
logicalNot() - Method in class ai.djl.engine.rust.RsNDArray
logicalNot(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
logicalOr(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
logicalOr(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
logicalXor(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
logicalXor(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
logSoftmax(int) - Method in class ai.djl.engine.rust.RsNDArray
logSoftmax(long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
lpPool(float, Shape, Shape, Shape, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
lpPool(long, float, long[], long[], boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
lstm(NDArray, NDList, NDList, boolean, int, double, boolean, boolean, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
lt(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
lt(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
lt(Number) - Method in class ai.djl.engine.rust.RsNDArray
lte(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
lte(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
lte(Number) - Method in class ai.djl.engine.rust.RsNDArray

M

matmul(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
matMul(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
max() - Method in class ai.djl.engine.rust.RsNDArray
max(int[], boolean) - Method in class ai.djl.engine.rust.RsNDArray
max(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
maximum(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
maximum(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
maximum(Number) - Method in class ai.djl.engine.rust.RsNDArray
maxPool(long, long[], long[], long[], boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
maxPool(Shape, Shape, Shape, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
maxWithAxis(long, int, boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
mean() - Method in class ai.djl.engine.rust.RsNDArray
mean(int[], boolean) - Method in class ai.djl.engine.rust.RsNDArray
mean(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
meanWithAxis(long, int[], boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
median() - Method in class ai.djl.engine.rust.RsNDArray
median(int[]) - Method in class ai.djl.engine.rust.RsNDArray
median(long, int, boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
min() - Method in class ai.djl.engine.rust.RsNDArray
min(int[], boolean) - Method in class ai.djl.engine.rust.RsNDArray
min(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
minimum(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
minimum(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
minimum(Number) - Method in class ai.djl.engine.rust.RsNDArray
minWithAxis(long, int, boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
mod(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
mod(Number) - Method in class ai.djl.engine.rust.RsNDArray
modi(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
modi(Number) - Method in class ai.djl.engine.rust.RsNDArray
mul(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
mul(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
mul(Number) - Method in class ai.djl.engine.rust.RsNDArray
muli(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
muli(Number) - Method in class ai.djl.engine.rust.RsNDArray
multiBoxDetection(NDList, boolean, float, int, float, boolean, int) - Method in class ai.djl.engine.rust.RsNDArrayEx
multiBoxPrior(List<Float>, List<Float>, List<Float>, List<Float>, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
multiBoxTarget(NDList, float, float, float, float, int) - Method in class ai.djl.engine.rust.RsNDArrayEx

N

nagUpdate(NDList, NDList, float, float, float, float, float) - Method in class ai.djl.engine.rust.RsNDArrayEx
neg() - Method in class ai.djl.engine.rust.RsNDArray
neg(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
negi() - Method in class ai.djl.engine.rust.RsNDArray
neq(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
neq(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
neq(Number) - Method in class ai.djl.engine.rust.RsNDArray
newBaseManager() - Method in class ai.djl.engine.rust.RsEngine
newBaseManager(Device) - Method in class ai.djl.engine.rust.RsEngine
newInstance(InputStream, Map<String, String>) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Create a pre-trained HuggingFaceTokenizer instance from InputStream.
newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.huggingface.translator.FillMaskTranslatorFactory
newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslatorFactory
newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.huggingface.translator.TextClassificationTranslatorFactory
newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslatorFactory
newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslatorFactory
newInstance(String) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Creates a pre-trained HuggingFaceTokenizer instance from huggingface hub.
newInstance(String, Map<String, String>) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Create a pre-trained HuggingFaceTokenizer instance from huggingface hub.
newInstance(Path) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Create a pre-trained HuggingFaceTokenizer instance from existing models.
newInstance(Path, Path, Map<String, String>) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Create a pre-trained BPE HuggingFaceTokenizer instance from existing models.
newInstance(Path, Map<String, String>) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
Create a pre-trained HuggingFaceTokenizer instance from existing models.
newModel(String, Device) - Method in class ai.djl.engine.rust.RsEngine
newSubManager(Device) - Method in class ai.djl.engine.rust.RsNDManager
none() - Method in class ai.djl.engine.rust.RsNDArray
nonzero() - Method in class ai.djl.engine.rust.RsNDArray
nonZero(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
norm(boolean) - Method in class ai.djl.engine.rust.RsNDArray
norm(int, int[], boolean) - Method in class ai.djl.engine.rust.RsNDArray
norm(long, int, int[], boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
normalize(double, long, double) - Method in class ai.djl.engine.rust.RsNDArray
normalize(long, double, long, double) - Static method in class ai.djl.engine.rust.RustLibrary
 

O

oneHot(int, float, float, DataType) - Method in class ai.djl.engine.rust.RsNDArray
oneHot(long, int, float, float, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
ones(long[], int, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
ones(Shape, DataType) - Method in class ai.djl.engine.rust.RsNDManager
optAddSpecialTokens(boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Sets if add special tokens.
optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.CrossEncoderTranslator.Builder
Sets the Batchifier for the Translator.
optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.FillMaskTranslator.Builder
Sets the Batchifier for the Translator.
optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator.Builder
Sets the Batchifier for the Translator.
optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.TextClassificationTranslator.Builder
Sets the Batchifier for the Translator.
optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator.Builder
Sets the Batchifier for the Translator.
optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator.Builder
Sets the Batchifier for the Translator.
optDense(String) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator.Builder
Sets the dense layer model file for the Translator.
optDenseActivation(String) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator.Builder
Sets the dense activation function for the Translator.
optDoLowerCase(boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Sets the doLowerCase for the tokenizer.
optDoLowerCase(String) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Sets the doLowerCase for the tokenizer with specific locale.
optIncludeTokenTypes(boolean) - Method in class ai.djl.huggingface.translator.CrossEncoderTranslator.Builder
Sets if include token types for the Translator.
optIncludeTokenTypes(boolean) - Method in class ai.djl.huggingface.translator.FillMaskTranslator.Builder
Sets if include token types for the Translator.
optIncludeTokenTypes(boolean) - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator.Builder
Sets if include token types for the Translator.
optIncludeTokenTypes(boolean) - Method in class ai.djl.huggingface.translator.TextClassificationTranslator.Builder
Sets if include token types for the Translator.
optIncludeTokenTypes(boolean) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator.Builder
Sets if include token types for the Translator.
optIncludeTokenTypes(boolean) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator.Builder
Sets if include token types for the Translator.
optLayerNorm(String) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator.Builder
Sets the LayerNorm model for the Translator.
optManager(NDManager) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Sets the optional manager used to manage the lifecycle of the tokenizer.
optMaskToken(String) - Method in class ai.djl.huggingface.translator.FillMaskTranslator.Builder
Sets the id of the mask Translator.
optMaxLength(int) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Sets maxLength for padding and truncation.
optNormalize(boolean) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator.Builder
Sets the normalize for the Translator.
optPadding(boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Enables or Disables default padding behavior for the tokenizer.
optPadToMaxLength() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Enables padding to pad sequences to previously specified maxLength, or modelMaxLength if not specified.
optPadToMultipleOf(int) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Sets padToMultipleOf for padding.
optPoolingMode(String) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator.Builder
Sets the pooling for the Translator.
optSigmoid(boolean) - Method in class ai.djl.huggingface.translator.CrossEncoderTranslator.Builder
Sets if apply sigmoid for the Translator.
optSoftmax(boolean) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator.Builder
Sets if implement softmax operation for the Translator.
optStride(int) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Sets the stride to use in overflow overlap when truncating sequences longer than the model supports.
optTokenizerName(String) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Sets the name of the tokenizer.
optTokenizerPath(Path) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Sets the file path of the tokenizer.
optTopK(int) - Method in class ai.djl.huggingface.translator.FillMaskTranslator.Builder
Set the topK number of classes to be displayed.
optTruncateFirstOnly() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Enables truncation to only truncate the first item.
optTruncateSecondOnly() - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Enables truncation to only truncate the second item.
optTruncation(boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Enables or Disables default truncation behavior for the tokenizer.
optWithOverflowingTokens(boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer.Builder
Sets if add special tokens.

P

percentile(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
percentile(Number) - Method in class ai.djl.engine.rust.RsNDArray
percentile(Number, int[]) - Method in class ai.djl.engine.rust.RsNDArray
percentileWithAxes(long, double, int[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
permute(long, int[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
pick(long, long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
pow(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
pow(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
pow(Number) - Method in class ai.djl.engine.rust.RsNDArray
powi(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
powi(Number) - Method in class ai.djl.engine.rust.RsNDArray
prelu(NDArray, NDArray) - Method in class ai.djl.engine.rust.RsNDArrayEx
prepare(TranslatorContext) - Method in class ai.djl.huggingface.translator.TextClassificationTranslator
prepare(TranslatorContext) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator
prepare(TranslatorContext) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator
PretrainedConfig - Class in ai.djl.huggingface.translator
A class representing HuggingFace config.json file.
PretrainedConfig() - Constructor for class ai.djl.huggingface.translator.PretrainedConfig
 
processInput(TranslatorContext, QAInput) - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator
processInput(TranslatorContext, StringPair) - Method in class ai.djl.huggingface.translator.CrossEncoderTranslator
processInput(TranslatorContext, String) - Method in class ai.djl.huggingface.translator.FillMaskTranslator
processInput(TranslatorContext, String) - Method in class ai.djl.huggingface.translator.TextClassificationTranslator
processInput(TranslatorContext, String) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator
processInput(TranslatorContext, String) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.CrossEncoderTranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.FillMaskTranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.TextClassificationTranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator
prod() - Method in class ai.djl.engine.rust.RsNDArray
prod(int[], boolean) - Method in class ai.djl.engine.rust.RsNDArray
prod(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
put(long, long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
put(NDArray, NDArray) - Method in class ai.djl.engine.rust.RsNDArray

Q

QuestionAnsweringTranslator - Class in ai.djl.huggingface.translator
The translator for Huggingface question answering model.
QuestionAnsweringTranslator.Builder - Class in ai.djl.huggingface.translator
The builder for question answering translator.
QuestionAnsweringTranslatorFactory - Class in ai.djl.huggingface.translator
A TranslatorFactory that creates a QuestionAnsweringTranslator instance.
QuestionAnsweringTranslatorFactory() - Constructor for class ai.djl.huggingface.translator.QuestionAnsweringTranslatorFactory
 

R

randint(long, long, long[], int, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
randomBrightness(float) - Method in class ai.djl.engine.rust.RsNDArrayEx
randomColorJitter(float, float, float, float) - Method in class ai.djl.engine.rust.RsNDArrayEx
randomFlipLeftRight() - Method in class ai.djl.engine.rust.RsNDArrayEx
randomFlipTopBottom() - Method in class ai.djl.engine.rust.RsNDArrayEx
randomHue(float) - Method in class ai.djl.engine.rust.RsNDArrayEx
randomInteger(long, long, Shape, DataType) - Method in class ai.djl.engine.rust.RsNDManager
randomNormal(float, float, long[], int, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
randomNormal(float, float, Shape, DataType) - Method in class ai.djl.engine.rust.RsNDManager
randomPermutation(long) - Method in class ai.djl.engine.rust.RsNDManager
randomPermutation(long, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
randomUniform(float, float, Shape, DataType) - Method in class ai.djl.engine.rust.RsNDManager
rdiv(NDArray) - Method in class ai.djl.engine.rust.RsNDArrayEx
rdiv(Number) - Method in class ai.djl.engine.rust.RsNDArrayEx
rdivi(NDArray) - Method in class ai.djl.engine.rust.RsNDArrayEx
rdivi(Number) - Method in class ai.djl.engine.rust.RsNDArrayEx
real() - Method in class ai.djl.engine.rust.RsNDArray
real(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
relu() - Method in class ai.djl.engine.rust.RsNDArrayEx
relu(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
remainder(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
repeat(int, long) - Method in class ai.djl.engine.rust.RsNDArray
repeat(long) - Method in class ai.djl.engine.rust.RsNDArray
repeat(long[]) - Method in class ai.djl.engine.rust.RsNDArray
repeat(long, long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
repeat(Shape) - Method in class ai.djl.engine.rust.RsNDArray
reshape(long, long[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
reshape(Shape) - Method in class ai.djl.engine.rust.RsNDArray
resize(int, int, int) - Method in class ai.djl.engine.rust.RsNDArrayEx
returnResource(NDManager) - Method in class ai.djl.engine.rust.RsNDArray
rmod(NDArray) - Method in class ai.djl.engine.rust.RsNDArrayEx
rmod(Number) - Method in class ai.djl.engine.rust.RsNDArrayEx
rmodi(NDArray) - Method in class ai.djl.engine.rust.RsNDArrayEx
rmodi(Number) - Method in class ai.djl.engine.rust.RsNDArrayEx
rmspropUpdate(NDList, NDList, float, float, float, float, float, float, float, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
rnn(NDArray, NDArray, NDList, boolean, int, RNN.Activation, double, boolean, boolean, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
rot90(long, int, int[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
rotate90(int, int[]) - Method in class ai.djl.engine.rust.RsNDArray
round() - Method in class ai.djl.engine.rust.RsNDArray
round(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
rpow(Number) - Method in class ai.djl.engine.rust.RsNDArrayEx
rpowi(Number) - Method in class ai.djl.engine.rust.RsNDArrayEx
RsEngine - Class in ai.djl.engine.rust
The RsEngine is an implementation of the Engine rust engine.
RsEngineProvider - Class in ai.djl.engine.rust
RsEngineProvider is the Rust implementation of EngineProvider.
RsEngineProvider() - Constructor for class ai.djl.engine.rust.RsEngineProvider
 
RsModel - Class in ai.djl.engine.rust
RsModel is the Rust implementation of Model.
RsModelZoo - Class in ai.djl.engine.rust.zoo
RsModelZoo is a repository that contains HuggingFace models.
RsNDArray - Class in ai.djl.engine.rust
RsNDArray is the Rust implementation of NDArray.
RsNDArray(RsNDManager, long) - Constructor for class ai.djl.engine.rust.RsNDArray
Constructs a Rust NDArray from a native handle (internal.
RsNDArray(RsNDManager, long, DataType, ByteBuffer) - Constructor for class ai.djl.engine.rust.RsNDArray
Constructs a Rust NDArray from a native handle (internal.
RsNDArrayEx - Class in ai.djl.engine.rust
PtNDArrayEx is the Rust implementation of the NDArrayEx.
RsNDArrayIndexer - Class in ai.djl.engine.rust
The NDArrayIndexer used by the RsNDArray.
RsNDManager - Class in ai.djl.engine.rust
PtNDManager is the Rust implementation of NDManager.
RsSymbolBlock - Class in ai.djl.engine.rust
RsSymbolBlock is the Rust implementation of SymbolBlock.
RsSymbolBlock(RsNDManager, long) - Constructor for class ai.djl.engine.rust.RsSymbolBlock
Constructs a RsSymbolBlock.
rsub(NDArray) - Method in class ai.djl.engine.rust.RsNDArrayEx
rsub(Number) - Method in class ai.djl.engine.rust.RsNDArrayEx
rsubi(NDArray) - Method in class ai.djl.engine.rust.RsNDArrayEx
rsubi(Number) - Method in class ai.djl.engine.rust.RsNDArrayEx
RsZooProvider - Class in ai.djl.engine.rust.zoo
An Rust model zoo provider implements the ZooProvider interface.
RsZooProvider() - Constructor for class ai.djl.engine.rust.zoo.RsZooProvider
 
runInference(long, long[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
RustLibrary - Class in ai.djl.engine.rust
Rust native library.

S

scatter(long, long, long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
scatter(NDArray, NDArray, int) - Method in class ai.djl.engine.rust.RsNDArray
selu() - Method in class ai.djl.engine.rust.RsNDArrayEx
selu(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
sequenceMask(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
sequenceMask(NDArray, float) - Method in class ai.djl.engine.rust.RsNDArray
set(NDArray, NDIndexFullSlice, NDArray) - Method in class ai.djl.engine.rust.RsNDArrayIndexer
set(NDArray, NDIndexFullSlice, Number) - Method in class ai.djl.engine.rust.RsNDArrayIndexer
set(Buffer) - Method in class ai.djl.engine.rust.RsNDArray
setId2label(Map<String, String>) - Method in class ai.djl.huggingface.translator.PretrainedConfig
Sets the id to label map.
setName(String) - Method in class ai.djl.engine.rust.RsNDArray
setPadding(long, int, String, int) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
setProblemType(String) - Method in class ai.djl.huggingface.translator.PretrainedConfig
Sets the problem type.
setRequiresGradient(boolean) - Method in class ai.djl.engine.rust.RsNDArray
setTruncation(long, int, String, int) - Method in class ai.djl.huggingface.tokenizers.jni.TokenizersLibrary
 
sgdUpdate(NDList, NDList, float, float, float, float, float, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
sigmoid() - Method in class ai.djl.engine.rust.RsNDArrayEx
sigmoid(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
sign() - Method in class ai.djl.engine.rust.RsNDArray
sign(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
signi() - Method in class ai.djl.engine.rust.RsNDArray
sin() - Method in class ai.djl.engine.rust.RsNDArray
sin(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
sinh() - Method in class ai.djl.engine.rust.RsNDArray
sinh(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
softmax(int) - Method in class ai.djl.engine.rust.RsNDArray
softmax(long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
softPlus() - Method in class ai.djl.engine.rust.RsNDArrayEx
softPlus(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
softSign() - Method in class ai.djl.engine.rust.RsNDArrayEx
softSign(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
sort() - Method in class ai.djl.engine.rust.RsNDArray
sort(int) - Method in class ai.djl.engine.rust.RsNDArray
sort(long, int, boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
split(long[], int) - Method in class ai.djl.engine.rust.RsNDArray
split(long, long[], int) - Static method in class ai.djl.engine.rust.RustLibrary
 
sqrt() - Method in class ai.djl.engine.rust.RsNDArray
sqrt(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
square() - Method in class ai.djl.engine.rust.RsNDArray
square(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
squeeze(int[]) - Method in class ai.djl.engine.rust.RsNDArray
squeeze(long, int[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
stack(long[], int) - Static method in class ai.djl.engine.rust.RustLibrary
 
stack(NDList, int) - Method in class ai.djl.engine.rust.RsNDArrayEx
stft(long, long, boolean, NDArray, boolean, boolean) - Method in class ai.djl.engine.rust.RsNDArray
stopGradient() - Method in class ai.djl.engine.rust.RsNDArray
sub(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
sub(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
sub(Number) - Method in class ai.djl.engine.rust.RsNDArray
subi(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
subi(Number) - Method in class ai.djl.engine.rust.RsNDArray
sum() - Method in class ai.djl.engine.rust.RsNDArray
sum(int[], boolean) - Method in class ai.djl.engine.rust.RsNDArray
sum(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
sumWithAxis(long, int[], boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
swapAxes(int, int) - Method in class ai.djl.engine.rust.RsNDArray

T

take(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
take(NDManager, NDArray) - Method in class ai.djl.engine.rust.RsNDArray
tan() - Method in class ai.djl.engine.rust.RsNDArray
tan(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
tanh() - Method in class ai.djl.engine.rust.RsNDArray
tanh() - Method in class ai.djl.engine.rust.RsNDArrayEx
tanh(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
tempAttach(NDManager) - Method in class ai.djl.engine.rust.RsNDArray
tensorOf(ByteBuffer, long[], int, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
TextClassificationTranslator - Class in ai.djl.huggingface.translator
The translator for Huggingface text classification model.
TextClassificationTranslator.Builder - Class in ai.djl.huggingface.translator
The builder for token classification translator.
TextClassificationTranslatorFactory - Class in ai.djl.huggingface.translator
A TranslatorFactory that creates a TextClassificationTranslator instance.
TextClassificationTranslatorFactory() - Constructor for class ai.djl.huggingface.translator.TextClassificationTranslatorFactory
 
TextEmbeddingTranslator - Class in ai.djl.huggingface.translator
The translator for Huggingface text embedding model.
TextEmbeddingTranslator.Builder - Class in ai.djl.huggingface.translator
The builder for token classification translator.
TextEmbeddingTranslatorFactory - Class in ai.djl.huggingface.translator
A TranslatorFactory that creates a TextEmbeddingTranslator instance.
TextEmbeddingTranslatorFactory() - Constructor for class ai.djl.huggingface.translator.TextEmbeddingTranslatorFactory
 
tile(int, long) - Method in class ai.djl.engine.rust.RsNDArray
tile(long) - Method in class ai.djl.engine.rust.RsNDArray
tile(long[]) - Method in class ai.djl.engine.rust.RsNDArray
tile(long, long[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
tile(Shape) - Method in class ai.djl.engine.rust.RsNDArray
tileWithAxis(long, int, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
tileWithShape(long, long[]) - Static method in class ai.djl.engine.rust.RustLibrary
 
toBoolean(long) - Static method in class ai.djl.engine.rust.RustLibrary
 
toByteBuffer(boolean) - Method in class ai.djl.engine.rust.RsNDArray
toDataType(long, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
toDegrees() - Method in class ai.djl.engine.rust.RsNDArray
toDense() - Method in class ai.djl.engine.rust.RsNDArray
toDevice(long, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
toDevice(Device, boolean) - Method in class ai.djl.engine.rust.RsNDArray
TokenClassificationTranslator - Class in ai.djl.huggingface.translator
The translator for Huggingface token classification model.
TokenClassificationTranslator.Builder - Class in ai.djl.huggingface.translator
The builder for token classification translator.
TokenClassificationTranslatorFactory - Class in ai.djl.huggingface.translator
A TranslatorFactory that creates a TokenClassificationTranslator instance.
TokenClassificationTranslatorFactory() - Constructor for class ai.djl.huggingface.translator.TokenClassificationTranslatorFactory
 
tokenize(String) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
TokenizersLibrary - Class in ai.djl.huggingface.tokenizers.jni
A class containing utilities to interact with the Tokenizer JNI layer.
toNDList(NDManager, boolean) - Method in class ai.djl.huggingface.tokenizers.Encoding
Returns the NDList representation of the encoding.
topK(int, int, boolean, boolean) - Method in class ai.djl.engine.rust.RsNDArray
topK(long, int, int, boolean, boolean) - Static method in class ai.djl.engine.rust.RustLibrary
 
toRadians() - Method in class ai.djl.engine.rust.RsNDArray
toSparse(SparseFormat) - Method in class ai.djl.engine.rust.RsNDArray
toString() - Method in class ai.djl.engine.rust.RsEngine
toString() - Method in class ai.djl.engine.rust.RsNDArray
toStringArray(Charset) - Method in class ai.djl.engine.rust.RsNDArray
toType(DataType, boolean) - Method in class ai.djl.engine.rust.RsNDArray
trace(int, int, int) - Method in class ai.djl.engine.rust.RsNDArray
transpose() - Method in class ai.djl.engine.rust.RsNDArray
transpose(int...) - Method in class ai.djl.engine.rust.RsNDArray
transpose(long, int, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
trunc() - Method in class ai.djl.engine.rust.RsNDArray
trunc(long) - Static method in class ai.djl.engine.rust.RustLibrary
 

U

uniform(float, float, long[], int, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
unique(Integer, boolean, boolean, boolean) - Method in class ai.djl.engine.rust.RsNDArray

W

where(long, long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
where(NDArray, NDArray) - Method in class ai.djl.engine.rust.RsNDArrayEx

X

xlogy(long, long) - Static method in class ai.djl.engine.rust.RustLibrary
 
xlogy(NDArray) - Method in class ai.djl.engine.rust.RsNDArray

Z

zeros(long[], int, String, int) - Static method in class ai.djl.engine.rust.RustLibrary
 
zeros(Shape, DataType) - Method in class ai.djl.engine.rust.RsNDManager
A B C D E F G H I L M N O P Q R S T U W X Z 
All Classes and Interfaces|All Packages|Constant Field Values|Serialized Form