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 java.lang.Object
implements ai.djl.ndarray.internal.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)
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	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.NDList inputs, int numItems, int embeddingSize, boolean sparseGrad, ai.djl.ndarray.types.DataType dataType, ai.djl.util.PairList<java.lang.String,java.lang.Object> additional)
ai.djl.ndarray.NDArray	gelu()
ai.djl.ndarray.NDArray	getArray()
ai.djl.ndarray.index.NDArrayIndexer	getIndexer()
ai.djl.ndarray.NDArray	globalAvgPool()
ai.djl.ndarray.NDArray	globalLpPool(float normType)
ai.djl.ndarray.NDArray	globalMaxPool()
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.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)
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(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.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	rdiv(ai.djl.ndarray.NDArray b)
ai.djl.ndarray.NDArray	rdiv(java.lang.Number n)
ai.djl.ndarray.NDArray	rdivi(ai.djl.ndarray.NDArray b)
ai.djl.ndarray.NDArray	rdivi(java.lang.Number n)
ai.djl.ndarray.NDArray	relu()
ai.djl.ndarray.NDArray	resize(int width, int height, int interpolation)
ai.djl.ndarray.NDArray	rmod(ai.djl.ndarray.NDArray b)
ai.djl.ndarray.NDArray	rmod(java.lang.Number n)
ai.djl.ndarray.NDArray	rmodi(ai.djl.ndarray.NDArray b)
ai.djl.ndarray.NDArray	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.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)
ai.djl.ndarray.NDArray	rpow(java.lang.Number n)
ai.djl.ndarray.NDArray	rpowi(java.lang.Number n)
ai.djl.ndarray.NDArray	rsub(ai.djl.ndarray.NDArray b)
ai.djl.ndarray.NDArray	rsub(java.lang.Number n)
ai.djl.ndarray.NDArray	rsubi(ai.djl.ndarray.NDArray b)
ai.djl.ndarray.NDArray	rsubi(java.lang.Number n)
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, swish

Method Detail

rdiv

public ai.djl.ndarray.NDArray rdiv(java.lang.Number n)

Specified by:

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

rdiv

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

Specified by:

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

rdivi

public ai.djl.ndarray.NDArray rdivi(java.lang.Number n)

Specified by:

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

rdivi

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

Specified by:

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

rsub

public ai.djl.ndarray.NDArray rsub(java.lang.Number n)

Specified by:

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

rsub

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

Specified by:

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

rsubi

public ai.djl.ndarray.NDArray rsubi(java.lang.Number n)

Specified by:

rsubi 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

rmod

public ai.djl.ndarray.NDArray rmod(java.lang.Number n)

Specified by:

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

rmod

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

Specified by:

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

rmodi

public ai.djl.ndarray.NDArray rmodi(java.lang.Number n)

Specified by:

rmodi 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

rpow

public ai.djl.ndarray.NDArray rpow(java.lang.Number n)

Specified by:

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

rpowi

public ai.djl.ndarray.NDArray rpowi(java.lang.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 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

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.NDList inputs,
                                       int numItems,
                                       int embeddingSize,
                                       boolean sparseGrad,
                                       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.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.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

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

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()

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(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 ai.djl.ndarray.NDArray getArray()

Specified by:

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