Package ai.djl.tensorflow.engine

Class TfNDArrayEx

java.lang.Object
ai.djl.tensorflow.engine.TfNDArrayEx
All Implemented Interfaces:
ai.djl.ndarray.internal.NDArrayEx
public class TfNDArrayEx extends Object implements ai.djl.ndarray.internal.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)
    ai.djl.ndarray.NDArray
    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)
    ai.djl.ndarray.NDArray
    concat(ai.djl.ndarray.NDList arrays, 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)
    ai.djl.ndarray.NDArray
    elu(float alpha)
    ai.djl.ndarray.NDList
    embedding(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDArray weight, ai.djl.ndarray.types.SparseFormat sparse)
    ai.djl.ndarray.NDArray
    gelu()
    ai.djl.ndarray.NDArray
    getArray()
    ai.djl.ndarray.index.NDArrayIndexer
    getIndexer(ai.djl.ndarray.NDManager manager)
    ai.djl.ndarray.NDArray
    globalAvgPool()
    ai.djl.ndarray.NDArray
    globalLpPool(float normType)
    ai.djl.ndarray.NDArray
    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)
    ai.djl.ndarray.NDArray
    leakyRelu(float alpha)
    ai.djl.ndarray.NDList
    linear(ai.djl.ndarray.NDArray input, ai.djl.ndarray.NDArray weight, ai.djl.ndarray.NDArray bias)
    ai.djl.ndarray.NDArray
    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)
    ai.djl.ndarray.NDArray
    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.NDArray
    normalize(float[] mean, float[] std)
    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)
    ai.djl.ndarray.NDArray
    rdivi(ai.djl.ndarray.NDArray b)
    ai.djl.ndarray.NDArray
    relu()
    ai.djl.ndarray.NDArray
    resize(int width, int height, int interpolation)
    ai.djl.ndarray.NDArray
    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 train, boolean bidirectional, boolean batchFirst)
    ai.djl.ndarray.NDArray
    rpowi(Number n)
    ai.djl.ndarray.NDArray
    rsubi(ai.djl.ndarray.NDArray b)
    ai.djl.ndarray.NDArray
    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)
    ai.djl.ndarray.NDArray
    sigmoid()
    ai.djl.ndarray.NDArray
    softPlus()
    ai.djl.ndarray.NDArray
    softSign()
    ai.djl.ndarray.NDArray
    stack(ai.djl.ndarray.NDList arrays)
    ai.djl.ndarray.NDArray
    stack(ai.djl.ndarray.NDList arrays, int axis)
    ai.djl.ndarray.NDArray
    tanh()
    ai.djl.ndarray.NDArray
    toTensor()
    ai.djl.ndarray.NDArray
    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, rdiv, rdiv, rdivi, rmod, rmod, rmodi, rpow, rsub, rsub, rsubi, swish

  • Method Details

    • rdivi

      public ai.djl.ndarray.NDArray rdivi(ai.djl.ndarray.NDArray b)
      Specified by:
      rdivi in interface ai.djl.ndarray.internal.NDArrayEx

    • rsubi

      public ai.djl.ndarray.NDArray rsubi(ai.djl.ndarray.NDArray b)
      Specified by:
      rsubi in interface ai.djl.ndarray.internal.NDArrayEx

    • rmodi

      public ai.djl.ndarray.NDArray rmodi(ai.djl.ndarray.NDArray b)
      Specified by:
      rmodi in interface ai.djl.ndarray.internal.NDArrayEx

    • rpowi

      public ai.djl.ndarray.NDArray rpowi(Number n)
      Specified by:
      rpowi in interface ai.djl.ndarray.internal.NDArrayEx

    • relu

      public ai.djl.ndarray.NDArray relu()
      Specified by:
      relu in interface ai.djl.ndarray.internal.NDArrayEx

    • sigmoid

      public ai.djl.ndarray.NDArray sigmoid()
      Specified by:
      sigmoid in interface ai.djl.ndarray.internal.NDArrayEx

    • tanh

      public ai.djl.ndarray.NDArray tanh()
      Specified by:
      tanh in interface ai.djl.ndarray.internal.NDArrayEx

    • softPlus

      public ai.djl.ndarray.NDArray softPlus()
      Specified by:
      softPlus in interface ai.djl.ndarray.internal.NDArrayEx

    • softSign

      public ai.djl.ndarray.NDArray softSign()
      Specified by:
      softSign in interface ai.djl.ndarray.internal.NDArrayEx

    • leakyRelu

      public ai.djl.ndarray.NDArray leakyRelu(float alpha)
      Specified by:
      leakyRelu in interface ai.djl.ndarray.internal.NDArrayEx

    • elu

      public ai.djl.ndarray.NDArray elu(float alpha)
      Specified by:
      elu in interface ai.djl.ndarray.internal.NDArrayEx

    • selu

      public ai.djl.ndarray.NDArray selu()
      Specified by:
      selu in interface ai.djl.ndarray.internal.NDArrayEx

    • gelu

      public ai.djl.ndarray.NDArray gelu()
      Specified by:
      gelu in interface ai.djl.ndarray.internal.NDArrayEx

    • maxPool

      public ai.djl.ndarray.NDArray 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 ai.djl.ndarray.NDArray globalMaxPool()
      Specified by:
      globalMaxPool in interface ai.djl.ndarray.internal.NDArrayEx

    • avgPool

      public ai.djl.ndarray.NDArray 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 ai.djl.ndarray.NDArray globalAvgPool()
      Specified by:
      globalAvgPool in interface ai.djl.ndarray.internal.NDArrayEx

    • lpPool

      public ai.djl.ndarray.NDArray 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 ai.djl.ndarray.NDArray 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 sparse)
      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 train, 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

    • normalize

      public ai.djl.ndarray.NDArray normalize(float[] mean, float[] std)
      Specified by:
      normalize in interface ai.djl.ndarray.internal.NDArrayEx

    • toTensor

      public ai.djl.ndarray.NDArray toTensor()
      Specified by:
      toTensor 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 ai.djl.ndarray.NDArray 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 ai.djl.ndarray.NDArray where(ai.djl.ndarray.NDArray condition, ai.djl.ndarray.NDArray other)
      Specified by:
      where in interface ai.djl.ndarray.internal.NDArrayEx

    • stack

      public ai.djl.ndarray.NDArray stack(ai.djl.ndarray.NDList arrays)
      Specified by:
      stack in interface ai.djl.ndarray.internal.NDArrayEx

    • stack

      public ai.djl.ndarray.NDArray stack(ai.djl.ndarray.NDList arrays, int axis)
      Specified by:
      stack in interface ai.djl.ndarray.internal.NDArrayEx

    • concat

      public ai.djl.ndarray.NDArray concat(ai.djl.ndarray.NDList arrays, 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 ai.djl.ndarray.NDArray getArray()
      Specified by:
      getArray in interface ai.djl.ndarray.internal.NDArrayEx