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	adadeltaUpdate(ai.djl.ndarray.NDList inputs, ai.djl.ndarray.NDList weights, float weightDecay, float rescaleGrad, float clipGrad, float rho, float epsilon)
void	adagradUpdate(ai.djl.ndarray.NDList inputs, ai.djl.ndarray.NDList weights, float learningRate, float weightDecay, float rescaleGrad, float clipGrad, float epsilon)
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 kernelShape, ai.djl.ndarray.types.Shape stride, ai.djl.ndarray.types.Shape padding, boolean ceilMode, boolean countIncludePad)
ai.djl.ndarray.NDList	batchNorm(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDArray runningMean, ai.djl.ndarray.NDArray runningVar, ai.djl.ndarray.NDArray gamma, ai.djl.ndarray.NDArray beta, int axis, float momentum, float eps, boolean training)
PtNDArray	concat(ai.djl.ndarray.NDList list, int axis)
ai.djl.ndarray.NDList	convolution(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDArray weight, ai.djl.ndarray.NDArray bias, ai.djl.ndarray.types.Shape stride, ai.djl.ndarray.types.Shape padding, ai.djl.ndarray.types.Shape dilation, int groups)
ai.djl.ndarray.NDList	deconvolution(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDArray weight, ai.djl.ndarray.NDArray bias, ai.djl.ndarray.types.Shape stride, ai.djl.ndarray.types.Shape padding, ai.djl.ndarray.types.Shape outPadding, ai.djl.ndarray.types.Shape dilation, int groups)
ai.djl.ndarray.NDList	dropout(ai.djl.ndarray.NDArray input, float rate, boolean training)
PtNDArray	elu(float alpha)
ai.djl.ndarray.NDList	embedding(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDArray weight, ai.djl.ndarray.types.SparseFormat sparseFormat)
PtNDArray	gelu()
PtNDArray	getArray()
ai.djl.ndarray.index.NDArrayIndexer	getIndexer()
PtNDArray	globalAvgPool()
PtNDArray	globalLpPool(float normType)
PtNDArray	globalMaxPool()
ai.djl.ndarray.NDList	gru(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDArray state, ai.djl.ndarray.NDList params, boolean hasBiases, int numLayers, double dropRate, boolean training, boolean bidirectional, boolean batchFirst)
PtNDArray	leakyRelu(float alpha)
ai.djl.ndarray.NDList	linear(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDArray weight, ai.djl.ndarray.NDArray bias)
PtNDArray	lpPool(float normType, ai.djl.ndarray.types.Shape kernelShape, ai.djl.ndarray.types.Shape stride, ai.djl.ndarray.types.Shape padding, boolean ceilMode)
ai.djl.ndarray.NDList	lstm(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDList states, ai.djl.ndarray.NDList params, boolean hasBiases, int numLayers, double dropRate, boolean training, boolean bidirectional, boolean batchFirst)
PtNDArray	maxPool(ai.djl.ndarray.types.Shape kernelShape, ai.djl.ndarray.types.Shape stride, ai.djl.ndarray.types.Shape padding, boolean ceilMode)
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)
ai.djl.ndarray.NDList	prelu(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDArray alpha)
ai.djl.ndarray.NDArray	randomBrightness(float brightness)
ai.djl.ndarray.NDArray	randomColorJitter(float brightness, float contrast, float saturation, float hue)
ai.djl.ndarray.NDArray	randomFlipLeftRight()
ai.djl.ndarray.NDArray	randomFlipTopBottom()
ai.djl.ndarray.NDArray	randomHue(float hue)
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, int interpolation)
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)
void	rmspropUpdate(ai.djl.ndarray.NDList inputs, ai.djl.ndarray.NDList weights, float learningRate, float weightDecay, float rescaleGrad, float clipGrad, float rho, float momentum, float epsilon, boolean centered)
ai.djl.ndarray.NDList	rnn(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDArray state, ai.djl.ndarray.NDList params, boolean hasBiases, int numLayers, ai.djl.nn.recurrent.RNN.Activation activation, double dropRate, boolean training, boolean bidirectional, boolean batchFirst)
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	softPlus()
PtNDArray	softSign()
PtNDArray	stack(ai.djl.ndarray.NDList arrays, int axis)
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, 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

softPlus

public PtNDArray softPlus()

Specified by:

softPlus 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 kernelShape,
                         ai.djl.ndarray.types.Shape stride,
                         ai.djl.ndarray.types.Shape padding,
                         boolean ceilMode)

Specified by:

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

globalMaxPool

public PtNDArray globalMaxPool()

Specified by:

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

avgPool

public PtNDArray avgPool(ai.djl.ndarray.types.Shape kernelShape,
                         ai.djl.ndarray.types.Shape stride,
                         ai.djl.ndarray.types.Shape padding,
                         boolean ceilMode,
                         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(float normType,
                        ai.djl.ndarray.types.Shape kernelShape,
                        ai.djl.ndarray.types.Shape stride,
                        ai.djl.ndarray.types.Shape padding,
                        boolean ceilMode)

Specified by:

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

globalLpPool

public PtNDArray globalLpPool(float normType)

Specified by:

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

adadeltaUpdate

public void adadeltaUpdate(ai.djl.ndarray.NDList inputs,
                           ai.djl.ndarray.NDList weights,
                           float weightDecay,
                           float rescaleGrad,
                           float clipGrad,
                           float rho,
                           float epsilon)

Specified by:

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

adagradUpdate

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

Specified by:

adagradUpdate 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

rmspropUpdate

public void rmspropUpdate(ai.djl.ndarray.NDList inputs,
                          ai.djl.ndarray.NDList weights,
                          float learningRate,
                          float weightDecay,
                          float rescaleGrad,
                          float clipGrad,
                          float rho,
                          float momentum,
                          float epsilon,
                          boolean centered)

Specified by:

rmspropUpdate 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.NDArray input,
                                         ai.djl.ndarray.NDArray weight,
                                         ai.djl.ndarray.NDArray bias,
                                         ai.djl.ndarray.types.Shape stride,
                                         ai.djl.ndarray.types.Shape padding,
                                         ai.djl.ndarray.types.Shape dilation,
                                         int groups)

Specified by:

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

deconvolution

public ai.djl.ndarray.NDList deconvolution(ai.djl.ndarray.NDArray input,
                                           ai.djl.ndarray.NDArray weight,
                                           ai.djl.ndarray.NDArray bias,
                                           ai.djl.ndarray.types.Shape stride,
                                           ai.djl.ndarray.types.Shape padding,
                                           ai.djl.ndarray.types.Shape outPadding,
                                           ai.djl.ndarray.types.Shape dilation,
                                           int groups)

Specified by:

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

linear

public ai.djl.ndarray.NDList linear(ai.djl.ndarray.NDArray input,
                                    ai.djl.ndarray.NDArray weight,
                                    ai.djl.ndarray.NDArray bias)

Specified by:

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

embedding

public ai.djl.ndarray.NDList embedding(ai.djl.ndarray.NDArray input,
                                       ai.djl.ndarray.NDArray weight,
                                       ai.djl.ndarray.types.SparseFormat sparseFormat)

Specified by:

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

prelu

public ai.djl.ndarray.NDList prelu(ai.djl.ndarray.NDArray input,
                                   ai.djl.ndarray.NDArray alpha)

Specified by:

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

dropout

public ai.djl.ndarray.NDList dropout(ai.djl.ndarray.NDArray input,
                                     float rate,
                                     boolean training)

Specified by:

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

batchNorm

public ai.djl.ndarray.NDList batchNorm(ai.djl.ndarray.NDArray input,
                                       ai.djl.ndarray.NDArray runningMean,
                                       ai.djl.ndarray.NDArray runningVar,
                                       ai.djl.ndarray.NDArray gamma,
                                       ai.djl.ndarray.NDArray beta,
                                       int axis,
                                       float momentum,
                                       float eps,
                                       boolean training)

Specified by:

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

rnn

public ai.djl.ndarray.NDList rnn(ai.djl.ndarray.NDArray input,
                                 ai.djl.ndarray.NDArray state,
                                 ai.djl.ndarray.NDList params,
                                 boolean hasBiases,
                                 int numLayers,
                                 ai.djl.nn.recurrent.RNN.Activation activation,
                                 double dropRate,
                                 boolean training,
                                 boolean bidirectional,
                                 boolean batchFirst)

Specified by:

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

gru

public ai.djl.ndarray.NDList gru(ai.djl.ndarray.NDArray input,
                                 ai.djl.ndarray.NDArray state,
                                 ai.djl.ndarray.NDList params,
                                 boolean hasBiases,
                                 int numLayers,
                                 double dropRate,
                                 boolean training,
                                 boolean bidirectional,
                                 boolean batchFirst)

Specified by:

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

lstm

public ai.djl.ndarray.NDList lstm(ai.djl.ndarray.NDArray input,
                                  ai.djl.ndarray.NDList states,
                                  ai.djl.ndarray.NDList params,
                                  boolean hasBiases,
                                  int numLayers,
                                  double dropRate,
                                  boolean training,
                                  boolean bidirectional,
                                  boolean batchFirst)

Specified by:

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

resize

public PtNDArray resize(int width,
                        int height,
                        int interpolation)

Specified by:

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

randomFlipLeftRight

public ai.djl.ndarray.NDArray randomFlipLeftRight()

Specified by:

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

randomFlipTopBottom

public ai.djl.ndarray.NDArray randomFlipTopBottom()

Specified by:

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

randomBrightness

public ai.djl.ndarray.NDArray randomBrightness(float brightness)

Specified by:

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

randomHue

public ai.djl.ndarray.NDArray randomHue(float hue)

Specified by:

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

randomColorJitter

public ai.djl.ndarray.NDArray randomColorJitter(float brightness,
                                                float contrast,
                                                float saturation,
                                                float hue)

Specified by:

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

getIndexer

public ai.djl.ndarray.index.NDArrayIndexer getIndexer()

Specified by:

getIndexer 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