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

adamUpdate(NDList, NDList, float, float, float, float, float, float, float, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

arange(float, float, float, DataType, Device) - 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

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

avgPool(Shape, Shape, Shape, PoolingConvention, boolean) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

B

batchNorm(NDList, float, float, int, boolean, boolean, PairList<String, Object>) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

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

cast(DataType) - Method in class ai.djl.pytorch.engine.PtModel

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

cat(NDArray[], 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

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

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.PtModel

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.NativeResource

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(NDList, Shape, Shape, Shape, Shape, int, int, String, boolean, PairList<String, Object>) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

Creates an PtNDArray with the given Native Memory Pointer and attaches to this manager.

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

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

createCSR(Buffer, long[], long[], Shape, Device) - Method in class ai.djl.pytorch.engine.PtNDManager

createEmptyNdArray(PtNDManager, Shape, DataType, Device, SparseFormat) - Static method in class ai.djl.pytorch.jni.JniUtils

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

createMask(Predicate<Number>) - Method in class ai.djl.pytorch.engine.PtNDArray

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

createRowSparse(Buffer, Shape, long[], Shape, Device) - Method in class ai.djl.pytorch.engine.PtNDManager

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

D

deleteModule(PtSymbolBlock) - Static method in class ai.djl.pytorch.jni.JniUtils

deleteNdArray(Pointer) - Static method in class ai.djl.pytorch.jni.JniUtils

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

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

dropout(NDList, float, int[], PairList<String, Object>) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

E

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

embedding(NDList, int, int, DataType, PairList<String, Object>) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

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, Device) - 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

finalize() - Method in class ai.djl.pytorch.engine.PtPredictor

finalize() - Method in class ai.djl.pytorch.jni.NativeResource

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

flatten(PtNDArray, long, 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(ParameterStore, NDList, PairList<String, Object>) - Method in class ai.djl.pytorch.engine.PtSymbolBlock

forward(PtSymbolBlock, NDList) - Static method in class ai.djl.pytorch.jni.IValueUtils

Run the forward of PyTorch module.

fromDeviceType(int) - Static method in class ai.djl.pytorch.engine.PtDeviceType

Converts from an PyTorch device number to Device.

fullyConnected(NDList, long, boolean, boolean, PairList<String, Object>) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

G

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

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

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

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

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

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

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

getHandle() - Method in class ai.djl.pytorch.jni.NativeResource

Gets the Pointer to this resource.

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

getOutputShapes(NDManager, Shape[]) - Method in class ai.djl.pytorch.engine.PtSymbolBlock

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

getParameterShape(String, Shape[]) - Method in class ai.djl.pytorch.engine.PtSymbolBlock

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

getUid() - Method in class ai.djl.pytorch.jni.NativeResource

Gets the unique ID of this resource.

getValue() - Method in class ai.djl.pytorch.jni.Pointer

Returns the native peer of the pointer address.

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

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

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

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

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

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

handle - Variable in class ai.djl.pytorch.jni.NativeResource

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

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

I

initialize(NDManager, DataType, Shape...) - Method in class ai.djl.pytorch.engine.PtSymbolBlock

invoke(String, NDArray[], NDArray[], PairList<String, ?>) - Method in class ai.djl.pytorch.engine.PtNDManager

invoke(String, NDList, PairList<String, ?>) - Method in class ai.djl.pytorch.engine.PtNDManager

isArray(Pointer) - Static method in class ai.djl.pytorch.jni.IValueUtils

Check IValue is a container of IValue Array.

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

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

isMap(Pointer) - 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

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

Check IValue is a container of PtNDArray.

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

Check IValue is a container of NDList.

isReleased() - Method in class ai.djl.pytorch.jni.NativeResource

Gets the boolean that indicates whether this resource has been released.

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

Check IValue is a container of String.

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

Check IValue is a container of IValue Tuple.

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

LibUtils - Class in ai.djl.pytorch.jni

Utilities for finding the PyTorch Engine binary on the System.

linspace(float, float, int, boolean, Device) - 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, String>) - Method in class ai.djl.pytorch.engine.PtModel

load(Path, Device) - Method in class ai.djl.pytorch.engine.PtNDManager

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

loadModule(PtNDManager, Path, Device) - 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

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

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

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

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

lpPool(Shape, Shape, Shape, PoolingConvention, int) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

lstm(NDList, long, float, int, boolean, boolean, boolean, double, double, PairList<String, Object>) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

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

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

NativeResource - Class in ai.djl.pytorch.jni

NativeResource is an internal class for AutoCloseable blocks of memory created in the PyTorch Engine.

NativeResource(Pointer) - Constructor for class ai.djl.pytorch.jni.NativeResource

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

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

newPredictor(Translator<I, O>) - Method in class ai.djl.pytorch.engine.PtModel

newSubManager() - Method in class ai.djl.pytorch.engine.PtNDManager

newSubManager(Device) - Method in class ai.djl.pytorch.engine.PtNDManager

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

normal(PtNDManager, double, double, Shape, DataType, Device) - Static method in class ai.djl.pytorch.jni.JniUtils

O

ones(Shape, DataType, Device) - 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(NDArray, int, boolean, String) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

Pointer - Class in ai.djl.pytorch.jni

An abstraction for a native pointer data type.

Pointer(long) - Constructor for class ai.djl.pytorch.jni.Pointer

Creates an instance of Pointer.

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(NDList, PairList<String, Object>) - 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

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

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.

PtNDArrayEx - Class in ai.djl.pytorch.engine

PtNDArrayEx is the PyTorch implementation of the NDArrayEx.

PtNDManager - Class in ai.djl.pytorch.engine

PtNDManager is the PyTorch implementation of NDManager.

PtPredictor<I,O> - Class in ai.djl.pytorch.engine

PtPredictor is the Pytorch implementation of Predictor.

PtSymbolBlock - Class in ai.djl.pytorch.engine

PtSymbolBlock is the PyTorch implementation of SymbolBlock.

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

Constructs a PtSymbolBlock.

R

randomMultinomial(int, NDArray) - Method in class ai.djl.pytorch.engine.PtNDManager

randomMultinomial(int, NDArray, Shape) - Method in class ai.djl.pytorch.engine.PtNDManager

randomNormal(float, float, Shape, DataType, Device) - Method in class ai.djl.pytorch.engine.PtNDManager

randomUniform(float, float, Shape, DataType, Device) - 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

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

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

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

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

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

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

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

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

rnn(NDList, String, long, float, int, boolean, boolean, boolean, PairList<String, Object>) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

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

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

set(NDIndex, Number) - Method in class ai.djl.pytorch.engine.PtNDArray

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

setInitializer(Initializer, String) - Method in class ai.djl.pytorch.engine.PtSymbolBlock

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

setScalar(NDIndex, Number) - 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

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

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

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

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

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

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

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(NDArray[], int) - 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

sumPool(Shape, Shape, Shape, PoolingConvention) - Method in class ai.djl.pytorch.engine.PtNDArrayEx

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

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

to(PtNDArray, DataType, Device, boolean) - 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

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(Pointer) - Static method in class ai.djl.pytorch.jni.IValueUtils

Extract IValue to an IValue Array.

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

Extract IValue to a Map.

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

Create IValue Pointer from NDArray.

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

Extract IValue with a PtNDArray value.

toNDList(Pointer, 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

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

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

Extract IValue to String.

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

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

W

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

Z

zeros(Shape, DataType, Device) - 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