A B C D E F G H I J L M N O P R S T U W Z

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

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

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

addi(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

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(int, int, int, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager

arange(float, float, float, 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(Number, Number) - Method in class ai.djl.pytorch.engine.PtNDArray

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

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

concat(NDList, int) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

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

contentEquals(NDArray) - 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(Number) - Method in class ai.djl.pytorch.engine.PtNDArray

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

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

divi(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

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

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

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

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

Run the forward of PyTorch module.

forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.pytorch.engine.PtSymbolBlock

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

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

getLibName() - 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

getVersion() - Method in class ai.djl.pytorch.engine.PtEngine

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

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

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

gte(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

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

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

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

isList(long) - Static method in class ai.djl.pytorch.jni.IValueUtils

Check IValue is a container of IValue List.

isMap(long) - Static method in class ai.djl.pytorch.jni.IValueUtils

Check IValue is a container of IValue Map.

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

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

isNDArray(long) - Static method in class ai.djl.pytorch.jni.IValueUtils

Check IValue is a container of PtNDArray.

isNDList(long) - Static method in class ai.djl.pytorch.jni.IValueUtils

Check IValue is a container of NDList.

isString(long) - Static method in class ai.djl.pytorch.jni.IValueUtils

Check IValue is a container of String.

isTuple(long) - Static method in class ai.djl.pytorch.jni.IValueUtils

Check IValue is a container of IValue Tuple.

iValueFromDict(long[], String[]) - Static method in class ai.djl.pytorch.jni.IValueUtils

Create Dict IValue Pointer from pointer with its key name.

iValueFromList(long[]) - Static method in class ai.djl.pytorch.jni.IValueUtils

Create List IValue Pointer from pointer array.

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

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

load(Path, String, Map<String, ?>) - Method in class ai.djl.pytorch.engine.PtModel

load(InputStream) - Method in class ai.djl.pytorch.engine.PtModel

Load PyTorch model from InputStream.

loadLibrary() - Static method in class ai.djl.pytorch.jni.LibUtils

loadModule(PtNDManager, Path, Device, String[], String[]) - Static method in class ai.djl.pytorch.jni.JniUtils

loadModule(PtNDManager, InputStream, Device, boolean) - Static method in class ai.djl.pytorch.jni.JniUtils

loadModuleHandle(InputStream, Device, 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(Number) - Method in class ai.djl.pytorch.engine.PtNDArray

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

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

lte(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

M

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

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

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

max(int[], boolean) - Method in class ai.djl.pytorch.engine.PtNDArray

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

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

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

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

maximum(NDArray) - 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, PtNDArray) - Static method in class ai.djl.pytorch.jni.JniUtils

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

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

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

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

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

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

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

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

moduleGetParams(PtSymbolBlock, PtNDManager) - Static method in class ai.djl.pytorch.jni.JniUtils

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

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

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

muli(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

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

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

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

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, float, float, DataType) - Method in class ai.djl.pytorch.engine.PtNDArray

ones(Shape, DataType) - Method in class ai.djl.pytorch.engine.PtNDManager

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

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

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

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

powi(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

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.

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

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, long) - Constructor for class ai.djl.pytorch.engine.PtSymbolBlock

Constructs a PtSymbolBlock.

PtSymbolBlock(PtNDManager) - Constructor for class ai.djl.pytorch.engine.PtSymbolBlock

Constructs an Empty 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(Number) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

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

rdivi(NDArray) - 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(long) - Method in class ai.djl.pytorch.engine.PtNDArray

repeat(int, 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(Number) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

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

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

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

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

rsubi(NDArray) - 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, float) - Method in class ai.djl.pytorch.engine.PtNDArray

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

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

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

set(NDArray, NDIndexBooleans, 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

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

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

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

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

subi(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

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(long) - 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(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

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.

toIValueArray(long) - Static method in class ai.djl.pytorch.jni.IValueUtils

Extract IValue to an IValue Array.

toIValueMap(long) - Static method in class ai.djl.pytorch.jni.IValueUtils

Extract IValue to a Map.

toIValuePointer(long) - Static method in class ai.djl.pytorch.jni.IValueUtils

Create IValue Pointer from NDArray.

toNDArray(long, PtNDManager) - Static method in class ai.djl.pytorch.jni.IValueUtils

Extract IValue with a PtNDArray value.

toNDList(long, PtNDManager) - Static method in class ai.djl.pytorch.jni.IValueUtils

Extract IValue to NDList.

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

toString(long) - Static method in class ai.djl.pytorch.jni.IValueUtils

Extract IValue to String.

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

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

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

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