Package 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 Object implements ai.djl.ndarray.internal.NDArrayEx
PtNDArrayEx is the PyTorch implementation of the NDArrayEx.

  • Method Summary

    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.NDArray
    interpolation(long[] size, int mode, boolean alignCorners)
    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(List<Float> sizes, List<Float> ratios, List<Float> steps, List<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
    rdivi(ai.djl.ndarray.NDArray b)
    PtNDArray
    relu()
    PtNDArray
    resize(int width, int height, int interpolation)
    PtNDArray
    rmodi(ai.djl.ndarray.NDArray b)
    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
    rpowi(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, rdiv, rdiv, rdivi, rmod, rmod, rmodi, rpow, rsub, rsub, rsubi, rsubi, stack, swish, toTensor

  • Method Details

    • rdivi

      public PtNDArray rdivi(ai.djl.ndarray.NDArray b)
      Specified by:
      rdivi 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

    • rpowi

      public PtNDArray rpowi(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

    • interpolation

      public ai.djl.ndarray.NDArray interpolation(long[] size, int mode, boolean alignCorners)
      Specified by:
      interpolation 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(List<Float> sizes, List<Float> ratios, List<Float> steps, List<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