Index
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
Translatorclasses 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
Encodingof the input text pair in batch. - batchEncode(PairList<String, String>, boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof the input text pair in batch. - batchEncode(String[]) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof the input sentence in batch. - batchEncode(String[], boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof the input sentence in batch. - batchEncode(List<String>) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof the input sentence in batch. - batchEncode(List<String>, boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof 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
CharSpaninstance. - 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.
- conj() - Method in class ai.djl.engine.rust.RsNDArray
- 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
Encodingof the input sentence. - encode(String[]) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof the input sentences. - encode(String[], boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof the input sentences. - encode(String, boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof the input sentence. - encode(String, String) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof the input sentence. - encode(String, String, boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof the input sentence. - encode(List<String>) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof the input sentences. - encode(List<String>, boolean, boolean) - Method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Returns the
Encodingof 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
TranslatorFactorythat creates aFillMaskTranslatorinstance. - 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
- 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
ZooProviderinterface. - HfZooProvider() - Constructor for class ai.djl.huggingface.zoo.HfZooProvider
- HuggingFaceTokenizer - Class in ai.djl.huggingface.tokenizers
-
HuggingFaceTokenizeris a Huggingface tokenizer implementation of theTokenizerinterface that converts sentences into token. - HuggingFaceTokenizer.Builder - Class in ai.djl.huggingface.tokenizers
-
The builder for creating huggingface tokenizer.
I
- ifft(long, long) - Method in class ai.djl.engine.rust.RsNDArray
- ifft2(long[], long[]) - Method in class ai.djl.engine.rust.RsNDArray
- intern(NDArray) - Method in class ai.djl.engine.rust.RsNDArray
- interpolation(long[], int, boolean) - Method in class ai.djl.engine.rust.RsNDArrayEx
- inverse() - Method in class ai.djl.engine.rust.RsNDArray
- inverse(long) - Static method in class ai.djl.engine.rust.RustLibrary
- irfft(long, long) - Method in class ai.djl.engine.rust.RsNDArray
- 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, 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
HuggingFaceTokenizerinstance fromInputStream. - 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
HuggingFaceTokenizerinstance from huggingface hub. - newInstance(String, Map<String, String>) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Create a pre-trained
HuggingFaceTokenizerinstance from huggingface hub. - newInstance(Path) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Create a pre-trained
HuggingFaceTokenizerinstance from existing models. - newInstance(Path, Path, Map<String, String>) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Create a pre-trained BPE
HuggingFaceTokenizerinstance from existing models. - newInstance(Path, Map<String, String>) - Static method in class ai.djl.huggingface.tokenizers.HuggingFaceTokenizer
-
Create a pre-trained
HuggingFaceTokenizerinstance 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
Batchifierfor theTranslator. - optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.FillMaskTranslator.Builder
-
Sets the
Batchifierfor theTranslator. - optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.QuestionAnsweringTranslator.Builder
-
Sets the
Batchifierfor theTranslator. - optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.TextClassificationTranslator.Builder
-
Sets the
Batchifierfor theTranslator. - optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.TextEmbeddingTranslator.Builder
-
Sets the
Batchifierfor theTranslator. - optBatchifier(Batchifier) - Method in class ai.djl.huggingface.translator.TokenClassificationTranslator.Builder
-
Sets the
Batchifierfor theTranslator. - 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
normalizefor theTranslator. - 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
- pad(Shape, double) - Method in class ai.djl.engine.rust.RsNDArray
- 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
TranslatorFactorythat creates aQuestionAnsweringTranslatorinstance. - 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
- rfft(long, long) - 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
RsEngineis an implementation of theEnginerust engine. - RsEngineProvider - Class in ai.djl.engine.rust
-
RsEngineProvideris the Rust implementation ofEngineProvider. - RsEngineProvider() - Constructor for class ai.djl.engine.rust.RsEngineProvider
- RsModel - Class in ai.djl.engine.rust
-
RsModelis the Rust implementation ofModel. - RsModelZoo - Class in ai.djl.engine.rust.zoo
-
RsModelZoo is a repository that contains HuggingFace models.
- RsNDArray - Class in ai.djl.engine.rust
-
RsNDArrayis the Rust implementation ofNDArray. - RsNDArray(RsNDManager, long) - Constructor for class ai.djl.engine.rust.RsNDArray
-
Constructs a Rust
NDArrayfrom a native handle (internal. - RsNDArray(RsNDManager, long, DataType, ByteBuffer) - Constructor for class ai.djl.engine.rust.RsNDArray
-
Constructs a Rust
NDArrayfrom a native handle (internal. - RsNDArrayEx - Class in ai.djl.engine.rust
-
PtNDArrayExis the Rust implementation of theNDArrayEx. - RsNDArrayIndexer - Class in ai.djl.engine.rust
-
The
NDArrayIndexerused by theRsNDArray. - RsNDManager - Class in ai.djl.engine.rust
-
PtNDManageris the Rust implementation ofNDManager. - RsSymbolBlock - Class in ai.djl.engine.rust
-
RsSymbolBlockis the Rust implementation ofSymbolBlock. - 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
ZooProviderinterface. - 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
TranslatorFactorythat creates aTextClassificationTranslatorinstance. - 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
TranslatorFactorythat creates aTextEmbeddingTranslatorinstance. - 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
- toByteArray(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
TranslatorFactorythat creates aTokenClassificationTranslatorinstance. - 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
NDListrepresentation 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
All Classes and Interfaces|All Packages|Constant Field Values|Serialized Form