A B C D E F G H I J L M N O P R S T U W Z
All Classes All Packages
All Classes All Packages
All Classes All Packages
A
- abs() - Method in class ai.djl.pytorch.engine.PtNDArray
- abs(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- acos() - Method in class ai.djl.pytorch.engine.PtNDArray
- acos(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- acosh() - Method in class ai.djl.pytorch.engine.PtNDArray
- adadeltaUpdate(NDList, NDList, float, float, float, float, float) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- adagradUpdate(NDList, NDList, float, float, float, float, float) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- adamUpdate(NDList, NDList, float, float, float, float, float, float, float, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- adamUpdate(PtNDArray, PtNDArray, PtNDArray, PtNDArray, float, float, float, float, float, float, float) - Static method in class ai.djl.pytorch.jni.JniUtils
- adaptiveAvgPool(PtNDArray, Shape) - Static method in class ai.djl.pytorch.jni.JniUtils
- adaptiveMaxPool(PtNDArray, Shape) - Static method in class ai.djl.pytorch.jni.JniUtils
- add(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- add(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- add(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- addi(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- addi(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- addi(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- ai.djl.pytorch.engine - package ai.djl.pytorch.engine
-
Contains classes to interface with the underlying PyTorch Engine.
- ai.djl.pytorch.jni - package ai.djl.pytorch.jni
-
Contains classes to interface with the underlying PyTorch Engine.
- all() - Method in class ai.djl.pytorch.engine.PtNDArray
- all(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- allocateDirect(int) - Method in class ai.djl.pytorch.engine.PtNDManager
- any() - Method in class ai.djl.pytorch.engine.PtNDArray
- any(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- arange(float, float, float, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- arange(int, int, int, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- arange(PtNDManager, float, float, float, DataType, Device, SparseFormat) - Static method in class ai.djl.pytorch.jni.JniUtils
- argMax() - Method in class ai.djl.pytorch.engine.PtNDArray
- argMax(int) - Method in class ai.djl.pytorch.engine.PtNDArray
- argMax(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- argMax(PtNDArray, long, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- argMin() - Method in class ai.djl.pytorch.engine.PtNDArray
- argMin(int) - Method in class ai.djl.pytorch.engine.PtNDArray
- argMin(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- argMin(PtNDArray, long, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- argSort(int, boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- argSort(PtNDArray, long, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- asin() - Method in class ai.djl.pytorch.engine.PtNDArray
- asin(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- asinh() - Method in class ai.djl.pytorch.engine.PtNDArray
- atan() - Method in class ai.djl.pytorch.engine.PtNDArray
- atan(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- atanh() - Method in class ai.djl.pytorch.engine.PtNDArray
- attach(NDManager) - Method in class ai.djl.pytorch.engine.PtNDArray
- attachGradient(PtNDArray, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- avgPool(Shape, Shape, Shape, boolean, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- avgPool(PtNDArray, Shape, Shape, Shape, boolean, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
B
- backward(NDArray) - Method in class ai.djl.pytorch.engine.PtGradientCollector
- backward(PtNDArray, PtNDArray, boolean, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- batchDot(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- batchNorm(NDArray, NDArray, NDArray, NDArray, NDArray, int, float, float, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- batchNorm(PtNDArray, PtNDArray, PtNDArray, PtNDArray, PtNDArray, boolean, double, double) - Static method in class ai.djl.pytorch.jni.JniUtils
- booleanMask(NDArray, int) - Method in class ai.djl.pytorch.engine.PtNDArray
- booleanMask(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- booleanMaskSet(PtNDArray, PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- broadcast(Shape) - Method in class ai.djl.pytorch.engine.PtNDArray
- broadcast(PtNDArray, Shape) - Static method in class ai.djl.pytorch.jni.JniUtils
C
- cat(PtNDArray[], long) - Static method in class ai.djl.pytorch.jni.JniUtils
- cbrt() - Method in class ai.djl.pytorch.engine.PtNDArray
- ceil() - Method in class ai.djl.pytorch.engine.PtNDArray
- ceil(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- clip(PtNDArray, Number, Number) - Static method in class ai.djl.pytorch.jni.JniUtils
- clip(Number, Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- clone(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- close() - Method in class ai.djl.pytorch.engine.PtGradientCollector
- close() - Method in class ai.djl.pytorch.engine.PtNDArray
- close() - Method in class ai.djl.pytorch.engine.PtSymbolBlock
- close() - Method in class ai.djl.pytorch.jni.IValue
- concat(NDList, int) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- contentEqual(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- contentEquals(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- contentEquals(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- convolution(NDArray, NDArray, NDArray, Shape, Shape, Shape, int) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- convolution(PtNDArray, PtNDArray, PtNDArray, Shape, Shape, Shape, int) - Static method in class ai.djl.pytorch.jni.JniUtils
- copyTo(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- cos() - Method in class ai.djl.pytorch.engine.PtNDArray
- cos(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- cosh() - Method in class ai.djl.pytorch.engine.PtNDArray
- cosh(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- create(Shape, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- create(Buffer, Shape, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- createCoo(Buffer, long[][], Shape) - Method in class ai.djl.pytorch.engine.PtNDManager
- createEmptyNdArray(PtNDManager, Shape, DataType, Device, SparseFormat) - Static method in class ai.djl.pytorch.jni.JniUtils
- createNdFromByteBuffer(PtNDManager, ByteBuffer, Shape, DataType, SparseFormat, Device) - Static method in class ai.djl.pytorch.jni.JniUtils
- createOnesNdArray(PtNDManager, Shape, DataType, Device, SparseFormat) - Static method in class ai.djl.pytorch.jni.JniUtils
- createSparseCoo(PtNDArray, PtNDArray, Shape) - Static method in class ai.djl.pytorch.jni.JniUtils
- createZerosNdArray(PtNDManager, Shape, DataType, Device, SparseFormat) - Static method in class ai.djl.pytorch.jni.JniUtils
- cumSum() - Method in class ai.djl.pytorch.engine.PtNDArray
- cumSum(int) - Method in class ai.djl.pytorch.engine.PtNDArray
- cumSum(PtNDArray, long) - Static method in class ai.djl.pytorch.jni.JniUtils
D
- deconvolution(NDArray, NDArray, NDArray, Shape, Shape, Shape, Shape, int) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- deleteModule(long) - Static method in class ai.djl.pytorch.jni.JniUtils
- deleteNDArray(long) - Static method in class ai.djl.pytorch.jni.JniUtils
- describeInput() - Method in class ai.djl.pytorch.engine.PtSymbolBlock
- describeOutput() - Method in class ai.djl.pytorch.engine.PtSymbolBlock
- detach() - Method in class ai.djl.pytorch.engine.PtNDArray
- detachGradient(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- div(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- div(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- div(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- divi(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- divi(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- divi(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- dot(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- dot(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- dropout(NDArray, float, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- dropout(PtNDArray, double, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- duplicate() - Method in class ai.djl.pytorch.engine.PtNDArray
E
- elu(float) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- elu(PtNDArray, double) - Static method in class ai.djl.pytorch.jni.JniUtils
- embedding(NDArray, NDArray, SparseFormat) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- embedding(PtNDArray, PtNDArray, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- enableInferenceMode(PtSymbolBlock) - Static method in class ai.djl.pytorch.jni.JniUtils
- enableTrainingMode(PtSymbolBlock) - Static method in class ai.djl.pytorch.jni.JniUtils
- ENGINE_NAME - Static variable in class ai.djl.pytorch.engine.PtEngine
- eq(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- eq(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- eq(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- equals(Object) - Method in class ai.djl.pytorch.engine.PtNDArray
- erfinv() - Method in class ai.djl.pytorch.engine.PtNDArray
- erfinv(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- exp() - Method in class ai.djl.pytorch.engine.PtNDArray
- exp(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- expandDims(int) - Method in class ai.djl.pytorch.engine.PtNDArray
- eye(int, int, int, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- eye(PtNDManager, int, int, DataType, Device, SparseFormat) - Static method in class ai.djl.pytorch.jni.JniUtils
F
- flatten() - Method in class ai.djl.pytorch.engine.PtNDArray
- flatten(PtNDArray, long, long) - Static method in class ai.djl.pytorch.jni.JniUtils
- flip(int...) - Method in class ai.djl.pytorch.engine.PtNDArray
- flip(PtNDArray, long[]) - Static method in class ai.djl.pytorch.jni.JniUtils
- floor() - Method in class ai.djl.pytorch.engine.PtNDArray
- floor(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- forward(PtSymbolBlock, NDList, boolean) - Static method in class ai.djl.pytorch.jni.IValueUtils
-
Runs the forward of PyTorch module.
- forward(PtSymbolBlock, IValue...) - Static method in class ai.djl.pytorch.jni.IValueUtils
-
Runs the forward of PyTorch module.
- forward(IValue...) - Method in class ai.djl.pytorch.engine.PtSymbolBlock
-
Runs the forward of this PyTorch module.
- forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.pytorch.engine.PtSymbolBlock
- from(boolean) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typeboolean
. - from(double) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typedouble
. - from(long) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typelong
. - from(NDArray) - Method in class ai.djl.pytorch.engine.PtNDManager
- from(PtNDArray) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typePtNDArray
. - from(String) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typeString
. - fromDeviceType(int) - Static method in class ai.djl.pytorch.engine.PtDeviceType
-
Converts from an PyTorch device number to
Device
. - full(Shape, float, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- full(PtNDManager, Shape, double, DataType, Device, SparseFormat) - Static method in class ai.djl.pytorch.jni.JniUtils
G
- gelu() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- gelu(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- get(long...) - Method in class ai.djl.pytorch.engine.PtNDArray
- get(NDArray, NDIndexFullPick) - Method in class ai.djl.pytorch.engine.PtNDArrayIndexer
- get(NDArray, NDIndexFullSlice) - Method in class ai.djl.pytorch.engine.PtNDArrayIndexer
- getAlternativeEngine() - Method in class ai.djl.pytorch.engine.PtEngine
- getArray() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- getArtifactNames() - Method in class ai.djl.pytorch.engine.PtModel
- getByteBuffer(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- getDataType() - Method in class ai.djl.pytorch.engine.PtNDArray
- getDataType(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- getDevice() - Method in class ai.djl.pytorch.engine.PtNDArray
- getDevice(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- getEngine() - Method in class ai.djl.pytorch.engine.PtEngineProvider
- getEngine() - Method in class ai.djl.pytorch.engine.PtNDManager
- getEngineName() - Method in class ai.djl.pytorch.engine.PtEngine
- getEngineName() - Method in class ai.djl.pytorch.engine.PtEngineProvider
- getEngineRank() - Method in class ai.djl.pytorch.engine.PtEngineProvider
- getFeatures() - Static method in class ai.djl.pytorch.jni.JniUtils
- getGradient() - Method in class ai.djl.pytorch.engine.PtNDArray
- getGradient(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- getGradientFunctionNames(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- getHandle() - Method in class ai.djl.pytorch.engine.PtSymbolBlock
-
Get the native PyTorch model pointer.
- getIndexer() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- getItem(PtNDArray, long[]) - Static method in class ai.djl.pytorch.jni.JniUtils
- getLayout(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- getLibtorchPath() - Static method in class ai.djl.pytorch.jni.LibUtils
- getManager() - Method in class ai.djl.pytorch.engine.PtNDArray
- getName() - Method in class ai.djl.pytorch.engine.PtNDArray
- getNDArrayInternal() - Method in class ai.djl.pytorch.engine.PtNDArray
- getNumInteropThreads() - Static method in class ai.djl.pytorch.jni.JniUtils
- getNumThreads() - Static method in class ai.djl.pytorch.jni.JniUtils
- getOutputShapes(Shape[]) - Method in class ai.djl.pytorch.engine.PtSymbolBlock
- getRank() - Method in class ai.djl.pytorch.engine.PtEngine
- getShape() - Method in class ai.djl.pytorch.engine.PtNDArray
- getShape(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- getSparseFormat() - Method in class ai.djl.pytorch.engine.PtNDArray
- getSparseFormat(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- getType() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the type of the IValue.
- getVersion() - Method in class ai.djl.pytorch.engine.PtEngine
- getVersion() - Static method in class ai.djl.pytorch.jni.LibUtils
- globalAvgPool() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- globalLpPool(float) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- globalMaxPool() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- gru(NDArray, NDArray, NDList, boolean, int, double, boolean, boolean, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- gru(PtNDArray, PtNDArray, NDList, boolean, int, double, boolean, boolean, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- gt(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- gt(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- gt(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- gte(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- gte(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- gte(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
H
- hasCapability(String) - Method in class ai.djl.pytorch.engine.PtEngine
- hasGradient() - Method in class ai.djl.pytorch.engine.PtNDArray
- hashCode() - Method in class ai.djl.pytorch.engine.PtNDArray
I
- index(PtNDArray, long[], long[], long[]) - Static method in class ai.djl.pytorch.jni.JniUtils
- indexSet(PtNDArray, PtNDArray, long[], long[], long[]) - Static method in class ai.djl.pytorch.jni.JniUtils
- intern(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- interpolate(PtNDArray, long[], int, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- isBoolean() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
boolean
type. - isBooleanList() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
boolean[]
type. - isDouble() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
double
type. - isDoubleList() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
double[]
type. - isInf(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- isInfinite() - Method in class ai.djl.pytorch.engine.PtNDArray
- isList() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
IValue[]
type. - isLong() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
long
type. - isLongList() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
long[]
type. - isMap() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
Map<String, V>
type. - isNaN() - Method in class ai.djl.pytorch.engine.PtNDArray
- isNaN(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- isString() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
String
type. - isTensor() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
Tensor
type. - isTensorList() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a
IValue[]
type. - isTuple() - Method in class ai.djl.pytorch.jni.IValue
-
Returns if the IValue is a tuple type.
- IValue - Class in ai.djl.pytorch.jni
-
A class represent a PyTorch
IValue
data. - IValueUtils - Class in ai.djl.pytorch.jni
-
IValueUtils is utility class to deal with IValue in PyTorch.
J
- JniUtils - Class in ai.djl.pytorch.jni
-
A class containing utilities to interact with the PyTorch Engine's Java Native Interface (JNI) layer.
L
- layerNorm(NDArray, Shape, NDArray, NDArray, float) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- layerNorm(PtNDArray, Shape, PtNDArray, PtNDArray, double) - Static method in class ai.djl.pytorch.jni.JniUtils
- leakyRelu(float) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- leakyRelu(PtNDArray, double) - Static method in class ai.djl.pytorch.jni.JniUtils
- LibUtils - Class in ai.djl.pytorch.jni
-
Utilities for finding the PyTorch Engine binary on the System.
- linear(NDArray, NDArray, NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- linear(PtNDArray, PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- linspace(float, float, int, boolean) - Method in class ai.djl.pytorch.engine.PtNDManager
- linspace(PtNDManager, float, float, int, DataType, Device, SparseFormat) - Static method in class ai.djl.pytorch.jni.JniUtils
- listFrom(boolean...) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typeboolean[]
. - listFrom(double...) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typedouble[]
. - listFrom(long...) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typelong[]
. - listFrom(PtNDArray...) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typeNDArray[]
. - listFrom(IValue...) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typeNDArray[]
. - load(InputStream, boolean) - Method in class ai.djl.pytorch.engine.PtModel
-
Load PyTorch model from
InputStream
. - load(InputStream, Map<String, ?>) - Method in class ai.djl.pytorch.engine.PtModel
- load(Path, String, Map<String, ?>) - Method in class ai.djl.pytorch.engine.PtModel
- loadLibrary() - Static method in class ai.djl.pytorch.jni.LibUtils
- loadModule(PtNDManager, InputStream, boolean, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- loadModule(PtNDManager, Path, boolean, String[], String[]) - Static method in class ai.djl.pytorch.jni.JniUtils
- loadModuleHandle(InputStream, Device, boolean, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- loadParameters(NDManager, DataInputStream) - Method in class ai.djl.pytorch.engine.PtSymbolBlock
- log() - Method in class ai.djl.pytorch.engine.PtNDArray
- log(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- log10() - Method in class ai.djl.pytorch.engine.PtNDArray
- log10(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- log2() - Method in class ai.djl.pytorch.engine.PtNDArray
- log2(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- logicalAnd(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- logicalAnd(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- logicalNot() - Method in class ai.djl.pytorch.engine.PtNDArray
- logicalNot(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- logicalOr(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- logicalOr(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- logicalXor(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- logicalXor(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- logSoftmax(int) - Method in class ai.djl.pytorch.engine.PtNDArray
- logSoftmax(PtNDArray, long, DataType) - Static method in class ai.djl.pytorch.jni.JniUtils
- lpPool(float, Shape, Shape, Shape, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- lpPool(PtNDArray, double, Shape, Shape, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- lstm(NDArray, NDList, NDList, boolean, int, double, boolean, boolean, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- lstm(PtNDArray, NDList, NDList, boolean, int, double, boolean, boolean, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- lt(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- lt(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- lt(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- lte(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- lte(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- lte(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
M
- matmul(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- matMul(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- max() - Method in class ai.djl.pytorch.engine.PtNDArray
- max(int[], boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- max(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- max(PtNDArray, long, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- max(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- maximum(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- maximum(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- maxPool(Shape, Shape, Shape, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- maxPool(PtNDArray, Shape, Shape, Shape, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- mean() - Method in class ai.djl.pytorch.engine.PtNDArray
- mean(int[], boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- mean(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- mean(PtNDArray, long, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- median() - Method in class ai.djl.pytorch.engine.PtNDArray
- median(int[]) - Method in class ai.djl.pytorch.engine.PtNDArray
- min() - Method in class ai.djl.pytorch.engine.PtNDArray
- min(int[], boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- min(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- min(PtNDArray, long, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- min(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- minimum(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- minimum(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- mod(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- mod(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- modi(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- modi(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- moduleGetParams(PtSymbolBlock, PtNDManager) - Static method in class ai.djl.pytorch.jni.JniUtils
- mul(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- mul(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- mul(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- muli(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- muli(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- muli(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- multiBoxDetection(NDList, boolean, float, int, float, boolean, int) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- multiBoxPrior(List<Float>, List<Float>, List<Float>, List<Float>, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- multiBoxTarget(NDList, float, float, float, float, int) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
N
- nagUpdate(NDList, NDList, float, float, float, float, float) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- neg() - Method in class ai.djl.pytorch.engine.PtNDArray
- neg(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- negi() - Method in class ai.djl.pytorch.engine.PtNDArray
- negi(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- neq(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- neq(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- neq(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- newBaseManager() - Method in class ai.djl.pytorch.engine.PtEngine
- newBaseManager(Device) - Method in class ai.djl.pytorch.engine.PtEngine
- newGradientCollector() - Method in class ai.djl.pytorch.engine.PtEngine
- newModel(String, Device) - Method in class ai.djl.pytorch.engine.PtEngine
- newSubManager(Device) - Method in class ai.djl.pytorch.engine.PtNDManager
- newSymbolBlock(NDManager) - Method in class ai.djl.pytorch.engine.PtEngine
- newTrainer(TrainingConfig) - Method in class ai.djl.pytorch.engine.PtModel
- none() - Method in class ai.djl.pytorch.engine.PtNDArray
- none(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- nonzero() - Method in class ai.djl.pytorch.engine.PtNDArray
- nonZeros(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- norm(boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- norm(int, int[], boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- norm(PtNDArray, int, int[], boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- normal(PtNDManager, double, double, Shape, DataType, Device) - Static method in class ai.djl.pytorch.jni.JniUtils
O
- oneHot(int) - Method in class ai.djl.pytorch.engine.PtNDArray
- oneHot(int, float, float, DataType) - Method in class ai.djl.pytorch.engine.PtNDArray
- oneHot(int, DataType) - Method in class ai.djl.pytorch.engine.PtNDArray
- oneHot(PtNDArray, int, DataType) - Static method in class ai.djl.pytorch.jni.JniUtils
- ones(Shape, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- onesLike(PtNDArray, DataType, Device, SparseFormat) - Static method in class ai.djl.pytorch.jni.JniUtils
P
- percentile(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- percentile(Number, int[]) - Method in class ai.djl.pytorch.engine.PtNDArray
- permute(PtNDArray, long[]) - Static method in class ai.djl.pytorch.jni.JniUtils
- pick(PtNDArray, PtNDArray, long) - Static method in class ai.djl.pytorch.jni.JniUtils
- pow(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- pow(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- pow(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- powi(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- powi(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- powi(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- prelu(NDArray, NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- prod() - Method in class ai.djl.pytorch.engine.PtNDArray
- prod(int[], boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- prod(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- prod(PtNDArray, long, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- PtDeviceType - Class in ai.djl.pytorch.engine
-
DeviceType is the PyTorch equivalent of the types in
Device
. - PtEngine - Class in ai.djl.pytorch.engine
- PtEngineProvider - Class in ai.djl.pytorch.engine
-
PtEngineProvider
is the PyTorch implementation ofEngineProvider
. - PtEngineProvider() - Constructor for class ai.djl.pytorch.engine.PtEngineProvider
- PtGradientCollector - Class in ai.djl.pytorch.engine
-
PtGradientCollector
is the PyTorch implementation ofGradientCollector
. - PtGradientCollector() - Constructor for class ai.djl.pytorch.engine.PtGradientCollector
- PtModel - Class in ai.djl.pytorch.engine
-
PtModel
is the PyTorch implementation ofModel
. - PtNDArray - Class in ai.djl.pytorch.engine
-
PtNDArray
is the PyTorch implementation ofNDArray
. - PtNDArray(PtNDManager, long) - Constructor for class ai.djl.pytorch.engine.PtNDArray
-
Constructs a PyTorch
NDArray
from a native handle (internal. - PtNDArray(PtNDManager, long, ByteBuffer) - Constructor for class ai.djl.pytorch.engine.PtNDArray
-
Constructs a PyTorch
NDArray
from a native handle (internal. - PtNDArrayEx - Class in ai.djl.pytorch.engine
-
PtNDArrayEx
is the PyTorch implementation of theNDArrayEx
. - PtNDArrayIndexer - Class in ai.djl.pytorch.engine
-
The
NDArrayIndexer
used by thePtNDArray
. - PtNDManager - Class in ai.djl.pytorch.engine
-
PtNDManager
is the PyTorch implementation ofNDManager
. - PtSymbolBlock - Class in ai.djl.pytorch.engine
-
PtSymbolBlock
is the PyTorch implementation ofSymbolBlock
. - PtSymbolBlock(PtNDManager) - Constructor for class ai.djl.pytorch.engine.PtSymbolBlock
-
Constructs an Empty
PtSymbolBlock
. - PtSymbolBlock(PtNDManager, long) - Constructor for class ai.djl.pytorch.engine.PtSymbolBlock
-
Constructs a
PtSymbolBlock
.
R
- randint(PtNDManager, long, long, Shape, DataType, Device) - Static method in class ai.djl.pytorch.jni.JniUtils
- randomBrightness(float) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- randomColorJitter(float, float, float, float) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- randomFlipLeftRight() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- randomFlipTopBottom() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- randomHue(float) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- randomInteger(long, long, Shape, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- randomNormal(float, float, Shape, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- randomUniform(float, float, Shape, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- rdiv(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rdiv(Number) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rdivi(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rdivi(Number) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- relu() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- relu(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- remainder(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- remainderi(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- repeat(int, long) - Method in class ai.djl.pytorch.engine.PtNDArray
- repeat(long) - Method in class ai.djl.pytorch.engine.PtNDArray
- repeat(long[]) - Method in class ai.djl.pytorch.engine.PtNDArray
- repeat(Shape) - Method in class ai.djl.pytorch.engine.PtNDArray
- repeat(PtNDArray, long, long) - Static method in class ai.djl.pytorch.jni.JniUtils
- requiresGrad(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- reshape(Shape) - Method in class ai.djl.pytorch.engine.PtNDArray
- reshape(PtNDArray, long[]) - Static method in class ai.djl.pytorch.jni.JniUtils
- resize(int, int, int) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rmod(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rmod(Number) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rmodi(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rmodi(Number) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rmspropUpdate(NDList, NDList, float, float, float, float, float, float, float, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rnn(NDArray, NDArray, NDList, boolean, int, RNN.Activation, double, boolean, boolean, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rnn(PtNDArray, PtNDArray, NDList, boolean, int, RNN.Activation, double, boolean, boolean, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- rot90(PtNDArray, int, int[]) - Static method in class ai.djl.pytorch.jni.JniUtils
- rotate90(int, int[]) - Method in class ai.djl.pytorch.engine.PtNDArray
- round() - Method in class ai.djl.pytorch.engine.PtNDArray
- round(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- rpow(Number) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rpowi(Number) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rsub(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rsub(Number) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rsubi(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- rsubi(Number) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
S
- saveParameters(DataOutputStream) - Method in class ai.djl.pytorch.engine.PtSymbolBlock
- selu() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- selu(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- sequenceMask(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- sequenceMask(NDArray, float) - Method in class ai.djl.pytorch.engine.PtNDArray
- set(NDArray, NDIndexBooleans, NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayIndexer
- set(NDArray, NDIndexFullSlice, NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayIndexer
- set(NDArray, NDIndexFullSlice, Number) - Method in class ai.djl.pytorch.engine.PtNDArrayIndexer
- set(PtNDArray, ByteBuffer) - Static method in class ai.djl.pytorch.jni.JniUtils
- set(Buffer) - Method in class ai.djl.pytorch.engine.PtNDArray
- setGraphExecutorOptimize(boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- setName(String) - Method in class ai.djl.pytorch.engine.PtNDArray
- setNumInteropThreads(int) - Static method in class ai.djl.pytorch.jni.JniUtils
- setNumThreads(int) - Static method in class ai.djl.pytorch.jni.JniUtils
- setRandomSeed(int) - Method in class ai.djl.pytorch.engine.PtEngine
- setRequiresGradient(boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- setSeed(long) - Static method in class ai.djl.pytorch.jni.JniUtils
- sgdUpdate(NDList, NDList, float, float, float, float, float, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- sgdUpdate(PtNDArray, PtNDArray, PtNDArray, float, float, float, float, float) - Static method in class ai.djl.pytorch.jni.JniUtils
- sigmoid() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- sigmoid(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- sign() - Method in class ai.djl.pytorch.engine.PtNDArray
- sign(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- signi() - Method in class ai.djl.pytorch.engine.PtNDArray
- signi(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- sin() - Method in class ai.djl.pytorch.engine.PtNDArray
- sin(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- sinh() - Method in class ai.djl.pytorch.engine.PtNDArray
- sinh(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- slice(PtNDArray, long, long, long, long) - Static method in class ai.djl.pytorch.jni.JniUtils
- softmax(int) - Method in class ai.djl.pytorch.engine.PtNDArray
- softmax(PtNDArray, long, DataType) - Static method in class ai.djl.pytorch.jni.JniUtils
- softPlus() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- softPlus(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- softSign() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- softSign(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- sort() - Method in class ai.djl.pytorch.engine.PtNDArray
- sort(int) - Method in class ai.djl.pytorch.engine.PtNDArray
- sort(PtNDArray, long, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- split(long[], int) - Method in class ai.djl.pytorch.engine.PtNDArray
- split(long, int) - Method in class ai.djl.pytorch.engine.PtNDArray
- split(PtNDArray, long[], long) - Static method in class ai.djl.pytorch.jni.JniUtils
- split(PtNDArray, long, long) - Static method in class ai.djl.pytorch.jni.JniUtils
- sqrt() - Method in class ai.djl.pytorch.engine.PtNDArray
- sqrt(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- square() - Method in class ai.djl.pytorch.engine.PtNDArray
- square(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- squeeze() - Method in class ai.djl.pytorch.engine.PtNDArray
- squeeze(int) - Method in class ai.djl.pytorch.engine.PtNDArray
- squeeze(int[]) - Method in class ai.djl.pytorch.engine.PtNDArray
- squeeze(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- squeeze(PtNDArray, long) - Static method in class ai.djl.pytorch.jni.JniUtils
- stack(NDList, int) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- stack(PtNDArray[], int) - Static method in class ai.djl.pytorch.jni.JniUtils
- startProfile(boolean, boolean, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
-
Calls this method to start profile the area you are interested in.
- stopGradient() - Method in class ai.djl.pytorch.engine.PtNDArray
- stopProfile(String) - Static method in class ai.djl.pytorch.jni.JniUtils
- stringMapFrom(Map<String, PtNDArray>) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typeMap[String, PtNDArray]
. - sub(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- sub(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- sub(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- subi(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray
- subi(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- subi(Number) - Method in class ai.djl.pytorch.engine.PtNDArray
- sum() - Method in class ai.djl.pytorch.engine.PtNDArray
- sum(int[], boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- sum(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- sum(PtNDArray, long[], boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
- swapAxes(int, int) - Method in class ai.djl.pytorch.engine.PtNDArray
T
- tan() - Method in class ai.djl.pytorch.engine.PtNDArray
- tan(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- tanh() - Method in class ai.djl.pytorch.engine.PtNDArray
- tanh() - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- tanh(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- tempAttach(NDManager) - Method in class ai.djl.pytorch.engine.PtNDArray
- tile(int, long) - Method in class ai.djl.pytorch.engine.PtNDArray
- tile(long) - Method in class ai.djl.pytorch.engine.PtNDArray
- tile(long[]) - Method in class ai.djl.pytorch.engine.PtNDArray
- tile(Shape) - Method in class ai.djl.pytorch.engine.PtNDArray
- tile(PtNDArray, long[]) - Static method in class ai.djl.pytorch.jni.JniUtils
- to(PtNDArray, DataType, Device) - Static method in class ai.djl.pytorch.jni.JniUtils
- toBoolean() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
boolean
value of this IValue. - toBooleanArray() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
boolean[]
value of this IValue. - toByteBuffer() - Method in class ai.djl.pytorch.engine.PtNDArray
- toDegrees() - Method in class ai.djl.pytorch.engine.PtNDArray
- toDense() - Method in class ai.djl.pytorch.engine.PtNDArray
- toDense(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- toDevice(Device, boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- toDeviceType(Device) - Static method in class ai.djl.pytorch.engine.PtDeviceType
-
Converts a
Device
to the corresponding PyTorch device number. - toDouble() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
double
value of this IValue. - toDoubleArray() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
double[]
value of this IValue. - toIValueArray() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
IValue[]
value of this IValue list. - toIValueMap() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
Map<String, IValue>
value of this IValue. - toIValueTuple() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
Map<String, IValue>
value of this IValue. - toLong() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
long
value of this IValue. - toLongArray() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
long[]
value of this IValue. - toNDList(PtNDManager) - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
NDList
value of this IValue. - toRadians() - Method in class ai.djl.pytorch.engine.PtNDArray
- toSparse(SparseFormat) - Method in class ai.djl.pytorch.engine.PtNDArray
- toSparse(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- toString() - Method in class ai.djl.pytorch.engine.PtEngine
- toString() - Method in class ai.djl.pytorch.engine.PtNDArray
- toStringArray(Charset) - Method in class ai.djl.pytorch.engine.PtNDArray
- toStringValue() - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
String
value of this IValue. - toTensor(PtNDManager) - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
NDArray
value of this IValue. - toTensorArray(PtNDManager) - Method in class ai.djl.pytorch.jni.IValue
-
Returns the
NDArray[]
value of this IValue. - toType(DataType, boolean) - Method in class ai.djl.pytorch.engine.PtNDArray
- trace(int, int, int) - Method in class ai.djl.pytorch.engine.PtNDArray
- transpose() - Method in class ai.djl.pytorch.engine.PtNDArray
- transpose(int...) - Method in class ai.djl.pytorch.engine.PtNDArray
- transpose(PtNDArray, long, long) - Static method in class ai.djl.pytorch.jni.JniUtils
- trunc() - Method in class ai.djl.pytorch.engine.PtNDArray
- trunc(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- tupleFrom(IValue...) - Static method in class ai.djl.pytorch.jni.IValue
-
Creates a new
IValue
of typeNDArray[]
.
U
- uniform(PtNDManager, double, double, Shape, DataType, Device) - Static method in class ai.djl.pytorch.jni.JniUtils
- unsqueeze(PtNDArray, long) - Static method in class ai.djl.pytorch.jni.JniUtils
W
- where(NDArray, NDArray) - Method in class ai.djl.pytorch.engine.PtNDArrayEx
- where(PtNDArray, PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- writeModule(PtSymbolBlock, OutputStream, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils
Z
- zeroGrad(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils
- zeros(Shape, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager
- zerosLike(PtNDArray, DataType, Device, SparseFormat) - Static method in class ai.djl.pytorch.jni.JniUtils
All Classes All Packages