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

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 type boolean.

from(double) - Static method in class ai.djl.pytorch.jni.IValue

Creates a new IValue of type double.

from(long) - Static method in class ai.djl.pytorch.jni.IValue

Creates a new IValue of type long.

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 type PtNDArray.

from(String) - Static method in class ai.djl.pytorch.jni.IValue

Creates a new IValue of type String.

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

gather(NDArray, int) - Method in class ai.djl.pytorch.engine.PtNDArray

gather(PtNDArray, PtNDArray, long) - Static method in class ai.djl.pytorch.jni.JniUtils

gelu() - Method in class ai.djl.pytorch.engine.PtNDArrayEx

gelu(PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils

get(NDArray, NDIndexFullPick) - Method in class ai.djl.pytorch.engine.PtNDArrayIndexer

get(NDArray, NDIndexFullSlice) - Method in class ai.djl.pytorch.engine.PtNDArrayIndexer

get(NDArray, NDIndex) - Method in class ai.djl.pytorch.engine.PtNDArrayIndexer

get(NDManager, long...) - Method in class ai.djl.pytorch.engine.PtNDArray

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

getDirectParameters() - Method in class ai.djl.pytorch.engine.PtSymbolBlock

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(NDManager) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

getItem(PtNDArray, long[], PtNDManager) - 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[], PtNDManager) - Static method in class ai.djl.pytorch.jni.JniUtils

indexAdv(PtNDArray, NDIndex) - Static method in class ai.djl.pytorch.jni.JniUtils

indexAdvPut(PtNDArray, NDIndex, PtNDArray) - 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

inverse() - Method in class ai.djl.pytorch.engine.PtNDArray

inverse(PtNDArray) - 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 type boolean[].

listFrom(double...) - Static method in class ai.djl.pytorch.jni.IValue

Creates a new IValue of type double[].

listFrom(long...) - Static method in class ai.djl.pytorch.jni.IValue

Creates a new IValue of type long[].

listFrom(PtNDArray...) - Static method in class ai.djl.pytorch.jni.IValue

Creates a new IValue of type NDArray[].

listFrom(IValue...) - Static method in class ai.djl.pytorch.jni.IValue

Creates a new IValue of type NDArray[].

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

The PtEngine is an implementation of the Engine based on the PyTorch Deep Learning Framework.

PtEngineProvider - Class in ai.djl.pytorch.engine

PtEngineProvider is the PyTorch implementation of EngineProvider.

PtEngineProvider() - Constructor for class ai.djl.pytorch.engine.PtEngineProvider

PtGradientCollector - Class in ai.djl.pytorch.engine

PtGradientCollector is the PyTorch implementation of GradientCollector.

PtGradientCollector() - Constructor for class ai.djl.pytorch.engine.PtGradientCollector

PtModel - Class in ai.djl.pytorch.engine

PtModel is the PyTorch implementation of Model.

PtNDArray - Class in ai.djl.pytorch.engine

PtNDArray is the PyTorch implementation of NDArray.

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 the NDArrayEx.

PtNDArrayIndexer - Class in ai.djl.pytorch.engine

The NDArrayIndexer used by the PtNDArray.

PtNDManager - Class in ai.djl.pytorch.engine

PtNDManager is the PyTorch implementation of NDManager.

PtSymbolBlock - Class in ai.djl.pytorch.engine

PtSymbolBlock is the PyTorch implementation of SymbolBlock.

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.

put(NDArray, NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray

put(PtNDArray, PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils

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

returnResource(NDManager) - Method in class ai.djl.pytorch.engine.PtNDArray

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(NDArray, NDIndex, Object) - 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

setBenchmarkCuDNN(boolean) - Static method in class ai.djl.pytorch.jni.JniUtils

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

stringIValueMapFrom(Map<String, IValue>) - Static method in class ai.djl.pytorch.jni.IValue

Creates a new IValue of type Map[String, IValue].

stringMapFrom(Map<String, PtNDArray>) - Static method in class ai.djl.pytorch.jni.IValue

Creates a new IValue of type Map[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

take(NDArray) - Method in class ai.djl.pytorch.engine.PtNDArray

take(PtNDArray, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils

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 type NDArray[].

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

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