ai.djl.pytorch.engine

Class PtNDArrayEx

java.lang.Object

ai.djl.pytorch.engine.PtNDArrayEx

All Implemented Interfaces:

ai.djl.ndarray.internal.NDArrayEx

public class PtNDArrayEx
extends java.lang.Object
implements ai.djl.ndarray.internal.NDArrayEx

PtNDArrayEx is the PyTorch implementation of the NDArrayEx.

Method Summary

Modifier and Type	Method and Description
void	adamUpdate(ai.djl.ndarray.NDList inputs, ai.djl.ndarray.NDList weights, float learningRate, float weightDecay, float rescaleGrad, float clipGrad, float beta1, float beta2, float epsilon, boolean lazyUpdate)
PtNDArray	avgPool(ai.djl.ndarray.types.Shape kernel, ai.djl.ndarray.types.Shape stride, ai.djl.ndarray.types.Shape pad, ai.djl.nn.pooling.PoolingConvention poolingConvention, boolean countIncludePad)
ai.djl.ndarray.NDList	batchNorm(ai.djl.ndarray.NDList inputs, float epsilon, float momentum, int axis, boolean center, boolean scale, ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)
PtNDArray	concat(ai.djl.ndarray.NDList list, int axis)
ai.djl.ndarray.NDList	convolution(ai.djl.ndarray.NDList inputs, ai.djl.ndarray.types.Shape kernel, ai.djl.ndarray.types.Shape stride, ai.djl.ndarray.types.Shape pad, ai.djl.ndarray.types.Shape dilate, int numFilters, int numGroups, java.lang.String layout, boolean noBias, ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)
ai.djl.ndarray.NDList	dropout(ai.djl.ndarray.NDList inputs, float probability, int[] sharedAxes, ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)
PtNDArray	elu(float alpha)
ai.djl.ndarray.NDList	embedding(ai.djl.ndarray.NDList inputs, int numItems, int embeddingSize, ai.djl.ndarray.types.DataType dataType, ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)
ai.djl.ndarray.NDList	fullyConnected(ai.djl.ndarray.NDList inputs, long outChannels, boolean flatten, boolean noBias, ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)
PtNDArray	gelu()
PtNDArray	getArray()
PtNDArray	globalAvgPool()
PtNDArray	globalLpPool(int pValue)
PtNDArray	globalMaxPool()
PtNDArray	globalSumPool()
PtNDArray	leakyRelu(float alpha)
PtNDArray	lpPool(ai.djl.ndarray.types.Shape kernel, ai.djl.ndarray.types.Shape stride, ai.djl.ndarray.types.Shape pad, ai.djl.nn.pooling.PoolingConvention poolingConvention, int pValue)
ai.djl.ndarray.NDList	lstm(ai.djl.ndarray.NDList inputs, long stateSize, float dropRate, int numStackedLayers, boolean useSequenceLength, boolean useBidirectional, boolean stateOutputs, double lstmStateClipMin, double lstmStateClipMax, ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)
PtNDArray	maxPool(ai.djl.ndarray.types.Shape kernel, ai.djl.ndarray.types.Shape stride, ai.djl.ndarray.types.Shape pad, ai.djl.nn.pooling.PoolingConvention poolingConvention)
ai.djl.ndarray.NDList	multiBoxDetection(ai.djl.ndarray.NDList inputs, boolean clip, float threshold, int backgroundId, float nmsThreshold, boolean forceSuppress, int nmsTopK)
ai.djl.ndarray.NDList	multiBoxPrior(java.util.List<java.lang.Float> sizes, java.util.List<java.lang.Float> ratios, java.util.List<java.lang.Float> steps, java.util.List<java.lang.Float> offsets, boolean clip)
ai.djl.ndarray.NDList	multiBoxTarget(ai.djl.ndarray.NDList inputs, float iouThreshold, float ignoreLabel, float negativeMiningRatio, float negativeMiningThreshold, int minNegativeSamples)
void	nagUpdate(ai.djl.ndarray.NDList inputs, ai.djl.ndarray.NDList weights, float learningRate, float weightDecay, float rescaleGrad, float clipGrad, float momentum)
PtNDArray	pick(ai.djl.ndarray.NDArray index, int axis, boolean keepDims, java.lang.String mode)
ai.djl.ndarray.NDList	prelu(ai.djl.ndarray.NDList inputs, ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)
PtNDArray	rdiv(ai.djl.ndarray.NDArray b)
PtNDArray	rdiv(java.lang.Number n)
PtNDArray	rdivi(ai.djl.ndarray.NDArray b)
PtNDArray	rdivi(java.lang.Number n)
PtNDArray	relu()
PtNDArray	resize(int width, int height)
PtNDArray	rmod(ai.djl.ndarray.NDArray b)
PtNDArray	rmod(java.lang.Number n)
PtNDArray	rmodi(ai.djl.ndarray.NDArray b)
PtNDArray	rmodi(java.lang.Number n)
ai.djl.ndarray.NDList	rnn(ai.djl.ndarray.NDList inputs, java.lang.String mode, long stateSize, float dropRate, int numStackedLayers, boolean useSequenceLength, boolean useBidirectional, boolean stateOutputs, ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)
PtNDArray	rpow(java.lang.Number n)
PtNDArray	rpowi(java.lang.Number n)
PtNDArray	rsub(ai.djl.ndarray.NDArray b)
PtNDArray	rsub(java.lang.Number n)
PtNDArray	rsubi(ai.djl.ndarray.NDArray b)
PtNDArray	rsubi(java.lang.Number n)
PtNDArray	selu()
void	sgdUpdate(ai.djl.ndarray.NDList inputs, ai.djl.ndarray.NDList weights, float learningRate, float weightDecay, float rescaleGrad, float clipGrad, float momentum, boolean lazyUpdate)
PtNDArray	sigmoid()
PtNDArray	softrelu()
PtNDArray	softsign()
PtNDArray	stack(ai.djl.ndarray.NDList arrays, int axis)
PtNDArray	sumPool(ai.djl.ndarray.types.Shape kernel, ai.djl.ndarray.types.Shape stride, ai.djl.ndarray.types.Shape pad, ai.djl.nn.pooling.PoolingConvention poolingConvention)
PtNDArray	tanh()
PtNDArray	where(ai.djl.ndarray.NDArray condition, ai.djl.ndarray.NDArray other)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface ai.djl.ndarray.internal.NDArrayEx

concat, crop, mish, normalize, pick, stack, swish, toTensor

Method Detail

rdiv

public PtNDArray rdiv(java.lang.Number n)

Specified by:

rdiv in interface ai.djl.ndarray.internal.NDArrayEx

rdiv

public PtNDArray rdiv(ai.djl.ndarray.NDArray b)

Specified by:

rdiv in interface ai.djl.ndarray.internal.NDArrayEx

rdivi

public PtNDArray rdivi(java.lang.Number n)

Specified by:

rdivi in interface ai.djl.ndarray.internal.NDArrayEx

rdivi

public PtNDArray rdivi(ai.djl.ndarray.NDArray b)

Specified by:

rdivi in interface ai.djl.ndarray.internal.NDArrayEx

rsub

public PtNDArray rsub(java.lang.Number n)

Specified by:

rsub in interface ai.djl.ndarray.internal.NDArrayEx

rsub

public PtNDArray rsub(ai.djl.ndarray.NDArray b)

Specified by:

rsub in interface ai.djl.ndarray.internal.NDArrayEx

rsubi

public PtNDArray rsubi(java.lang.Number n)

Specified by:

rsubi in interface ai.djl.ndarray.internal.NDArrayEx

rsubi

public PtNDArray rsubi(ai.djl.ndarray.NDArray b)

Specified by:

rsubi in interface ai.djl.ndarray.internal.NDArrayEx

rmod

public PtNDArray rmod(java.lang.Number n)

Specified by:

rmod in interface ai.djl.ndarray.internal.NDArrayEx

rmod

public PtNDArray rmod(ai.djl.ndarray.NDArray b)

Specified by:

rmod in interface ai.djl.ndarray.internal.NDArrayEx

rmodi

public PtNDArray rmodi(java.lang.Number n)

Specified by:

rmodi in interface ai.djl.ndarray.internal.NDArrayEx

rmodi

public PtNDArray rmodi(ai.djl.ndarray.NDArray b)

Specified by:

rmodi in interface ai.djl.ndarray.internal.NDArrayEx

rpow

public PtNDArray rpow(java.lang.Number n)

Specified by:

rpow in interface ai.djl.ndarray.internal.NDArrayEx

rpowi

public PtNDArray rpowi(java.lang.Number n)

Specified by:

rpowi in interface ai.djl.ndarray.internal.NDArrayEx

relu

public PtNDArray relu()

Specified by:

relu in interface ai.djl.ndarray.internal.NDArrayEx

sigmoid

public PtNDArray sigmoid()

Specified by:

sigmoid in interface ai.djl.ndarray.internal.NDArrayEx

tanh

public PtNDArray tanh()

Specified by:

tanh in interface ai.djl.ndarray.internal.NDArrayEx

softrelu

public PtNDArray softrelu()

Specified by:

softrelu in interface ai.djl.ndarray.internal.NDArrayEx

softsign

public PtNDArray softsign()

Specified by:

softsign in interface ai.djl.ndarray.internal.NDArrayEx

leakyRelu

public PtNDArray leakyRelu(float alpha)

Specified by:

leakyRelu in interface ai.djl.ndarray.internal.NDArrayEx

elu

public PtNDArray elu(float alpha)

Specified by:

elu in interface ai.djl.ndarray.internal.NDArrayEx

selu

public PtNDArray selu()

Specified by:

selu in interface ai.djl.ndarray.internal.NDArrayEx

gelu

public PtNDArray gelu()

Specified by:

gelu in interface ai.djl.ndarray.internal.NDArrayEx

maxPool

public PtNDArray maxPool(ai.djl.ndarray.types.Shape kernel,
                         ai.djl.ndarray.types.Shape stride,
                         ai.djl.ndarray.types.Shape pad,
                         ai.djl.nn.pooling.PoolingConvention poolingConvention)

Specified by:

maxPool in interface ai.djl.ndarray.internal.NDArrayEx

globalMaxPool

public PtNDArray globalMaxPool()

Specified by:

globalMaxPool in interface ai.djl.ndarray.internal.NDArrayEx

sumPool

public PtNDArray sumPool(ai.djl.ndarray.types.Shape kernel,
                         ai.djl.ndarray.types.Shape stride,
                         ai.djl.ndarray.types.Shape pad,
                         ai.djl.nn.pooling.PoolingConvention poolingConvention)

Specified by:

sumPool in interface ai.djl.ndarray.internal.NDArrayEx

globalSumPool

public PtNDArray globalSumPool()

Specified by:

globalSumPool in interface ai.djl.ndarray.internal.NDArrayEx

avgPool

public PtNDArray avgPool(ai.djl.ndarray.types.Shape kernel,
                         ai.djl.ndarray.types.Shape stride,
                         ai.djl.ndarray.types.Shape pad,
                         ai.djl.nn.pooling.PoolingConvention poolingConvention,
                         boolean countIncludePad)

Specified by:

avgPool in interface ai.djl.ndarray.internal.NDArrayEx

globalAvgPool

public PtNDArray globalAvgPool()

Specified by:

globalAvgPool in interface ai.djl.ndarray.internal.NDArrayEx

lpPool

public PtNDArray lpPool(ai.djl.ndarray.types.Shape kernel,
                        ai.djl.ndarray.types.Shape stride,
                        ai.djl.ndarray.types.Shape pad,
                        ai.djl.nn.pooling.PoolingConvention poolingConvention,
                        int pValue)

Specified by:

lpPool in interface ai.djl.ndarray.internal.NDArrayEx

globalLpPool

public PtNDArray globalLpPool(int pValue)

Specified by:

globalLpPool in interface ai.djl.ndarray.internal.NDArrayEx

adamUpdate

public void adamUpdate(ai.djl.ndarray.NDList inputs,
                       ai.djl.ndarray.NDList weights,
                       float learningRate,
                       float weightDecay,
                       float rescaleGrad,
                       float clipGrad,
                       float beta1,
                       float beta2,
                       float epsilon,
                       boolean lazyUpdate)

Specified by:

adamUpdate in interface ai.djl.ndarray.internal.NDArrayEx

nagUpdate

public void nagUpdate(ai.djl.ndarray.NDList inputs,
                      ai.djl.ndarray.NDList weights,
                      float learningRate,
                      float weightDecay,
                      float rescaleGrad,
                      float clipGrad,
                      float momentum)

Specified by:

nagUpdate in interface ai.djl.ndarray.internal.NDArrayEx

sgdUpdate

public void sgdUpdate(ai.djl.ndarray.NDList inputs,
                      ai.djl.ndarray.NDList weights,
                      float learningRate,
                      float weightDecay,
                      float rescaleGrad,
                      float clipGrad,
                      float momentum,
                      boolean lazyUpdate)

Specified by:

sgdUpdate in interface ai.djl.ndarray.internal.NDArrayEx

convolution

public ai.djl.ndarray.NDList convolution(ai.djl.ndarray.NDList inputs,
                                         ai.djl.ndarray.types.Shape kernel,
                                         ai.djl.ndarray.types.Shape stride,
                                         ai.djl.ndarray.types.Shape pad,
                                         ai.djl.ndarray.types.Shape dilate,
                                         int numFilters,
                                         int numGroups,
                                         java.lang.String layout,
                                         boolean noBias,
                                         ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)

Specified by:

convolution in interface ai.djl.ndarray.internal.NDArrayEx

fullyConnected

public ai.djl.ndarray.NDList fullyConnected(ai.djl.ndarray.NDList inputs,
                                            long outChannels,
                                            boolean flatten,
                                            boolean noBias,
                                            ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)

Specified by:

fullyConnected in interface ai.djl.ndarray.internal.NDArrayEx

embedding

public ai.djl.ndarray.NDList embedding(ai.djl.ndarray.NDList inputs,
                                       int numItems,
                                       int embeddingSize,
                                       ai.djl.ndarray.types.DataType dataType,
                                       ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)

Specified by:

embedding in interface ai.djl.ndarray.internal.NDArrayEx

prelu

public ai.djl.ndarray.NDList prelu(ai.djl.ndarray.NDList inputs,
                                   ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)

Specified by:

prelu in interface ai.djl.ndarray.internal.NDArrayEx

dropout

public ai.djl.ndarray.NDList dropout(ai.djl.ndarray.NDList inputs,
                                     float probability,
                                     int[] sharedAxes,
                                     ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)

Specified by:

dropout in interface ai.djl.ndarray.internal.NDArrayEx

batchNorm

public ai.djl.ndarray.NDList batchNorm(ai.djl.ndarray.NDList inputs,
                                       float epsilon,
                                       float momentum,
                                       int axis,
                                       boolean center,
                                       boolean scale,
                                       ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)

Specified by:

batchNorm in interface ai.djl.ndarray.internal.NDArrayEx

rnn

public ai.djl.ndarray.NDList rnn(ai.djl.ndarray.NDList inputs,
                                 java.lang.String mode,
                                 long stateSize,
                                 float dropRate,
                                 int numStackedLayers,
                                 boolean useSequenceLength,
                                 boolean useBidirectional,
                                 boolean stateOutputs,
                                 ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)

Specified by:

rnn in interface ai.djl.ndarray.internal.NDArrayEx

lstm

public ai.djl.ndarray.NDList lstm(ai.djl.ndarray.NDList inputs,
                                  long stateSize,
                                  float dropRate,
                                  int numStackedLayers,
                                  boolean useSequenceLength,
                                  boolean useBidirectional,
                                  boolean stateOutputs,
                                  double lstmStateClipMin,
                                  double lstmStateClipMax,
                                  ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)

Specified by:

lstm in interface ai.djl.ndarray.internal.NDArrayEx

resize

public PtNDArray resize(int width,
                        int height)

Specified by:

resize in interface ai.djl.ndarray.internal.NDArrayEx

pick

public PtNDArray pick(ai.djl.ndarray.NDArray index,
                      int axis,
                      boolean keepDims,
                      java.lang.String mode)

Specified by:

pick in interface ai.djl.ndarray.internal.NDArrayEx

where

public PtNDArray where(ai.djl.ndarray.NDArray condition,
                       ai.djl.ndarray.NDArray other)

Specified by:

where in interface ai.djl.ndarray.internal.NDArrayEx

stack

public PtNDArray stack(ai.djl.ndarray.NDList arrays,
                       int axis)

Specified by:

stack in interface ai.djl.ndarray.internal.NDArrayEx

concat

public PtNDArray concat(ai.djl.ndarray.NDList list,
                        int axis)

Specified by:

concat in interface ai.djl.ndarray.internal.NDArrayEx

multiBoxTarget

public ai.djl.ndarray.NDList multiBoxTarget(ai.djl.ndarray.NDList inputs,
                                            float iouThreshold,
                                            float ignoreLabel,
                                            float negativeMiningRatio,
                                            float negativeMiningThreshold,
                                            int minNegativeSamples)

Specified by:

multiBoxTarget in interface ai.djl.ndarray.internal.NDArrayEx

multiBoxPrior

public ai.djl.ndarray.NDList multiBoxPrior(java.util.List<java.lang.Float> sizes,
                                           java.util.List<java.lang.Float> ratios,
                                           java.util.List<java.lang.Float> steps,
                                           java.util.List<java.lang.Float> offsets,
                                           boolean clip)

Specified by:

multiBoxPrior in interface ai.djl.ndarray.internal.NDArrayEx

multiBoxDetection

public ai.djl.ndarray.NDList multiBoxDetection(ai.djl.ndarray.NDList inputs,
                                               boolean clip,
                                               float threshold,
                                               int backgroundId,
                                               float nmsThreshold,
                                               boolean forceSuppress,
                                               int nmsTopK)

Specified by:

multiBoxDetection in interface ai.djl.ndarray.internal.NDArrayEx

getArray

public PtNDArray getArray()

Specified by:

getArray in interface ai.djl.ndarray.internal.NDArrayEx