Package ai.djl.pytorch.engine

Class 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

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method 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 learningRateBiasCorrection, float weightDecay, float rescaleGrad, float clipGrad, float beta1, float beta2, float epsilon, boolean lazyUpdate, boolean adamw)
      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​(ai.djl.ndarray.NDManager manager)
      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)
      ai.djl.ndarray.NDList layerNorm​(ai.djl.ndarray.NDArray input, ai.djl.ndarray.types.Shape normalizedShape, ai.djl.ndarray.NDArray gamma, ai.djl.ndarray.NDArray beta, float eps)
      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 learningRateBiasCorrection,
                       float weightDecay,
                       float rescaleGrad,
                       float clipGrad,
                       float beta1,
                       float beta2,
                       float epsilon,
                       boolean lazyUpdate,
                       boolean adamw)
        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

      • layerNorm

        public ai.djl.ndarray.NDList layerNorm​(ai.djl.ndarray.NDArray input,
                                       ai.djl.ndarray.types.Shape normalizedShape,
                                       ai.djl.ndarray.NDArray gamma,
                                       ai.djl.ndarray.NDArray beta,
                                       float eps)
        Specified by:
        layerNorm 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​(ai.djl.ndarray.NDManager manager)
        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