Class TfNDArray

  • All Implemented Interfaces:
    ai.djl.ndarray.BytesSupplier, ai.djl.ndarray.NDArray, ai.djl.ndarray.NDResource, java.lang.AutoCloseable

    public class TfNDArray
    extends ai.djl.util.NativeResource<org.tensorflow.internal.c_api.TFE_TensorHandle>
    implements ai.djl.ndarray.NDArray
    TfNDArray is the TensorFlow implementation of NDArray.
    • Field Summary

      • Fields inherited from class ai.djl.util.NativeResource

        handle
    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      ai.djl.ndarray.NDArray abs()
      ai.djl.ndarray.NDArray acos()
      ai.djl.ndarray.NDArray acosh()
      ai.djl.ndarray.NDArray add​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray add​(java.lang.Number n)
      ai.djl.ndarray.NDArray addi​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray addi​(java.lang.Number n)
      ai.djl.ndarray.NDArray all()
      ai.djl.ndarray.NDArray any()
      ai.djl.ndarray.NDArray argMax()
      ai.djl.ndarray.NDArray argMax​(int axis)
      ai.djl.ndarray.NDArray argMin()
      ai.djl.ndarray.NDArray argMin​(int axis)
      ai.djl.ndarray.NDArray argSort​(int axis, boolean ascending)
      ai.djl.ndarray.NDArray asin()
      ai.djl.ndarray.NDArray asinh()
      ai.djl.ndarray.NDArray atan()
      ai.djl.ndarray.NDArray atanh()
      void attach​(ai.djl.ndarray.NDManager manager)
      ai.djl.ndarray.NDArray batchDot​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray batchMatMul​(ai.djl.ndarray.NDArray other)  
      ai.djl.ndarray.NDArray booleanMask​(ai.djl.ndarray.NDArray index, int axis)
      ai.djl.ndarray.NDArray broadcast​(ai.djl.ndarray.types.Shape shape)
      ai.djl.ndarray.NDArray cbrt()
      ai.djl.ndarray.NDArray ceil()
      ai.djl.ndarray.NDArray clip​(java.lang.Number min, java.lang.Number max)
      void close()
      ai.djl.ndarray.NDArray complex()
      boolean contentEquals​(ai.djl.ndarray.NDArray other)
      boolean contentEquals​(java.lang.Number number)
      ai.djl.ndarray.NDArray cos()
      ai.djl.ndarray.NDArray cosh()
      ai.djl.ndarray.NDArray cumProd​(int axis)
      ai.djl.ndarray.NDArray cumProd​(int axis, ai.djl.ndarray.types.DataType dataType)
      ai.djl.ndarray.NDArray cumSum()
      ai.djl.ndarray.NDArray cumSum​(int axis)
      void detach()
      ai.djl.ndarray.NDArray div​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray div​(java.lang.Number n)
      ai.djl.ndarray.NDArray divi​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray divi​(java.lang.Number n)
      ai.djl.ndarray.NDArray dot​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray duplicate()
      ai.djl.ndarray.NDArray eq​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray eq​(java.lang.Number n)
      boolean equals​(java.lang.Object obj)
      ai.djl.ndarray.NDArray erfinv()
      ai.djl.ndarray.NDArray exp()
      ai.djl.ndarray.NDArray expandDims​(int axis)
      ai.djl.ndarray.NDArray fft​(long length, long axis)
      ai.djl.ndarray.NDArray flatten()
      ai.djl.ndarray.NDArray flatten​(int startDim, int endDim)
      ai.djl.ndarray.NDArray flip​(int... axes)
      ai.djl.ndarray.NDArray floor()
      ai.djl.ndarray.NDArray gammaln()
      ai.djl.ndarray.NDArray gather​(ai.djl.ndarray.NDArray index, int axis)
      ai.djl.ndarray.NDArray gatherNd​(ai.djl.ndarray.NDArray index)
      ai.djl.ndarray.types.DataType getDataType()
      ai.djl.Device getDevice()
      ai.djl.ndarray.NDArray getGradient()
      TfNDManager getManager()
      java.lang.String getName()
      ai.djl.ndarray.internal.NDArrayEx getNDArrayInternal()
      ai.djl.ndarray.types.Shape getShape()
      ai.djl.ndarray.types.SparseFormat getSparseFormat()
      ai.djl.ndarray.NDArray gt​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray gt​(java.lang.Number n)
      ai.djl.ndarray.NDArray gte​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray gte​(java.lang.Number n)
      boolean hasGradient()
      int hashCode()
      void intern​(ai.djl.ndarray.NDArray replaced)
      ai.djl.ndarray.NDArray inverse()
      ai.djl.ndarray.NDArray isInfinite()
      ai.djl.ndarray.NDArray isNaN()
      ai.djl.ndarray.NDArray log()
      ai.djl.ndarray.NDArray log10()
      ai.djl.ndarray.NDArray log2()
      ai.djl.ndarray.NDArray logicalAnd​(ai.djl.ndarray.NDArray n)
      ai.djl.ndarray.NDArray logicalNot()
      ai.djl.ndarray.NDArray logicalOr​(ai.djl.ndarray.NDArray n)
      ai.djl.ndarray.NDArray logicalXor​(ai.djl.ndarray.NDArray n)
      ai.djl.ndarray.NDArray logSoftmax​(int axis)
      ai.djl.ndarray.NDArray lt​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray lt​(java.lang.Number n)
      ai.djl.ndarray.NDArray lte​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray lte​(java.lang.Number n)
      ai.djl.ndarray.NDArray matMul​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray max()
      ai.djl.ndarray.NDArray max​(int[] axes, boolean keepDims)
      ai.djl.ndarray.NDArray maximum​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray maximum​(java.lang.Number n)
      ai.djl.ndarray.NDArray mean()
      ai.djl.ndarray.NDArray mean​(int[] axes, boolean keepDims)
      ai.djl.ndarray.NDArray median()
      ai.djl.ndarray.NDArray median​(int[] axes)
      ai.djl.ndarray.NDArray min()
      ai.djl.ndarray.NDArray min​(int[] axes, boolean keepDims)
      ai.djl.ndarray.NDArray minimum​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray minimum​(java.lang.Number n)
      ai.djl.ndarray.NDArray mod​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray mod​(java.lang.Number n)
      ai.djl.ndarray.NDArray modi​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray modi​(java.lang.Number n)
      ai.djl.ndarray.NDArray mul​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray mul​(java.lang.Number n)
      ai.djl.ndarray.NDArray muli​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray muli​(java.lang.Number n)
      ai.djl.ndarray.NDArray neg()
      ai.djl.ndarray.NDArray negi()
      ai.djl.ndarray.NDArray neq​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray neq​(java.lang.Number n)
      ai.djl.ndarray.NDArray nonzero()
      ai.djl.ndarray.NDArray norm​(boolean keepDims)
      ai.djl.ndarray.NDArray norm​(int ord, int[] axes, boolean keepDims)
      ai.djl.ndarray.NDArray normalize​(double p, long dim, double eps)
      ai.djl.ndarray.NDArray oneHot​(int depth, float onValue, float offValue, ai.djl.ndarray.types.DataType dataType)
      ai.djl.ndarray.NDArray percentile​(java.lang.Number percentile)
      ai.djl.ndarray.NDArray percentile​(java.lang.Number percentile, int[] dimension)
      ai.djl.ndarray.NDArray pow​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray pow​(java.lang.Number n)
      ai.djl.ndarray.NDArray powi​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray powi​(java.lang.Number n)
      ai.djl.ndarray.NDArray prod()
      ai.djl.ndarray.NDArray prod​(int[] axes, boolean keepDims)
      ai.djl.ndarray.NDArray put​(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value)
      ai.djl.ndarray.NDArray real()
      ai.djl.ndarray.NDArray repeat​(int axis, long repeats)
      ai.djl.ndarray.NDArray repeat​(long repeats)
      ai.djl.ndarray.NDArray repeat​(long[] repeats)
      ai.djl.ndarray.NDArray repeat​(ai.djl.ndarray.types.Shape desiredShape)
      ai.djl.ndarray.NDArray reshape​(ai.djl.ndarray.types.Shape shape)
      void returnResource​(ai.djl.ndarray.NDManager manager)
      ai.djl.ndarray.NDArray rotate90​(int times, int[] axes)
      ai.djl.ndarray.NDArray round()
      ai.djl.ndarray.NDArray scatter​(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value, int axis)
      ai.djl.ndarray.NDArray sequenceMask​(ai.djl.ndarray.NDArray sequenceLength)
      ai.djl.ndarray.NDArray sequenceMask​(ai.djl.ndarray.NDArray sequenceLength, float value)
      void set​(java.nio.Buffer buffer)
      protected void setHandle​(org.tensorflow.internal.c_api.TFE_TensorHandle newHandle)  
      void setName​(java.lang.String name)
      void setRequiresGradient​(boolean requiresGrad)
      ai.djl.ndarray.NDArray sign()
      ai.djl.ndarray.NDArray signi()
      ai.djl.ndarray.NDArray sin()
      ai.djl.ndarray.NDArray sinh()
      ai.djl.ndarray.NDArray softmax​(int axis)
      ai.djl.ndarray.NDArray sort()
      ai.djl.ndarray.NDArray sort​(int axis)
      ai.djl.ndarray.NDList split​(long[] indices, int axis)
      ai.djl.ndarray.NDArray sqrt()
      ai.djl.ndarray.NDArray square()
      ai.djl.ndarray.NDArray squeeze​(int[] axes)
      ai.djl.ndarray.NDArray stft​(long nFft, long hopLength, boolean center, ai.djl.ndarray.NDArray window, boolean normalize, boolean returnComplex)
      ai.djl.ndarray.NDArray stopGradient()
      ai.djl.ndarray.NDArray sub​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray sub​(java.lang.Number n)
      ai.djl.ndarray.NDArray subi​(ai.djl.ndarray.NDArray other)
      ai.djl.ndarray.NDArray subi​(java.lang.Number n)
      ai.djl.ndarray.NDArray sum()
      ai.djl.ndarray.NDArray sum​(int[] axes, boolean keepDims)
      ai.djl.ndarray.NDArray take​(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray index)
      ai.djl.ndarray.NDArray tan()
      ai.djl.ndarray.NDArray tanh()
      void tempAttach​(ai.djl.ndarray.NDManager manager)
      ai.djl.ndarray.NDArray tile​(int axis, long repeats)
      ai.djl.ndarray.NDArray tile​(long repeats)
      ai.djl.ndarray.NDArray tile​(long[] repeats)
      ai.djl.ndarray.NDArray tile​(ai.djl.ndarray.types.Shape desiredShape)
      java.nio.ByteBuffer toByteBuffer()
      ai.djl.ndarray.NDArray toDegrees()
      ai.djl.ndarray.NDArray toDense()
      ai.djl.ndarray.NDArray toDevice​(ai.djl.Device device, boolean copy)
      ai.djl.ndarray.NDArray toRadians()
      ai.djl.ndarray.NDArray toSparse​(ai.djl.ndarray.types.SparseFormat fmt)
      java.lang.String toString()
      java.lang.String[] toStringArray​(java.nio.charset.Charset charset)
      ai.djl.ndarray.NDArray toType​(ai.djl.ndarray.types.DataType dataType, boolean copy)
      ai.djl.ndarray.NDArray trace​(int offset, int axis1, int axis2)
      ai.djl.ndarray.NDArray transpose()
      ai.djl.ndarray.NDArray transpose​(int... dimensions)
      ai.djl.ndarray.NDArray trunc()
      ai.djl.ndarray.NDList unique​(java.lang.Integer dim, boolean sorted, boolean returnInverse, boolean returnCounts)
      ai.djl.ndarray.NDArray xlogy​(ai.djl.ndarray.NDArray other)  
      • Methods inherited from class ai.djl.util.NativeResource

        getHandle, getUid, isReleased, onClose
      • Methods inherited from class java.lang.Object

        clone, finalize, getClass, notify, notifyAll, wait, wait, wait
      • Methods inherited from interface ai.djl.ndarray.BytesSupplier

        getAsBytes, getAsObject, getAsString
      • Methods inherited from interface ai.djl.ndarray.NDArray

        allClose, allClose, argSort, argSort, booleanMask, broadcast, concat, concat, copyTo, countNonzero, countNonzero, encode, fft, get, get, get, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, getResourceNDArrays, getScalar, getUid, getUint8, isEmpty, isReleased, isScalar, isSparse, like, max, mean, min, none, norm, norm, norm, normalize, normalize, oneHot, oneHot, onesLike, prod, reshape, scaleGradient, set, set, set, set, set, set, set, set, set, setScalar, shapeEquals, size, size, split, split, split, squeeze, squeeze, stack, stack, stft, sum, swapAxes, take, toArray, toBooleanArray, toByteArray, toDebugString, toDebugString, toDebugString, toDoubleArray, toFloatArray, toIntArray, toLongArray, toStringArray, toUint8Array, trace, trace, unique, unique, zerosLike
    • Method Detail

      • getManager

        public TfNDManager getManager()
        Specified by:
        getManager in interface ai.djl.ndarray.NDResource
      • getName

        public java.lang.String getName()
        Specified by:
        getName in interface ai.djl.ndarray.NDArray
      • setName

        public void setName​(java.lang.String name)
        Specified by:
        setName in interface ai.djl.ndarray.NDArray
      • getDataType

        public ai.djl.ndarray.types.DataType getDataType()
        Specified by:
        getDataType in interface ai.djl.ndarray.NDArray
      • getDevice

        public ai.djl.Device getDevice()
        Specified by:
        getDevice in interface ai.djl.ndarray.NDArray
      • getShape

        public ai.djl.ndarray.types.Shape getShape()
        Specified by:
        getShape in interface ai.djl.ndarray.NDArray
      • getSparseFormat

        public ai.djl.ndarray.types.SparseFormat getSparseFormat()
        Specified by:
        getSparseFormat in interface ai.djl.ndarray.NDArray
      • toDevice

        public ai.djl.ndarray.NDArray toDevice​(ai.djl.Device device,
                                               boolean copy)
        Specified by:
        toDevice in interface ai.djl.ndarray.NDArray
      • toType

        public ai.djl.ndarray.NDArray toType​(ai.djl.ndarray.types.DataType dataType,
                                             boolean copy)
        Specified by:
        toType in interface ai.djl.ndarray.NDArray
      • setRequiresGradient

        public void setRequiresGradient​(boolean requiresGrad)
        Specified by:
        setRequiresGradient in interface ai.djl.ndarray.NDArray
      • getGradient

        public ai.djl.ndarray.NDArray getGradient()
        Specified by:
        getGradient in interface ai.djl.ndarray.NDArray
      • hasGradient

        public boolean hasGradient()
        Specified by:
        hasGradient in interface ai.djl.ndarray.NDArray
      • stopGradient

        public ai.djl.ndarray.NDArray stopGradient()
        Specified by:
        stopGradient in interface ai.djl.ndarray.NDArray
      • toStringArray

        public java.lang.String[] toStringArray​(java.nio.charset.Charset charset)
        Specified by:
        toStringArray in interface ai.djl.ndarray.NDArray
      • toByteBuffer

        public java.nio.ByteBuffer toByteBuffer()
        Specified by:
        toByteBuffer in interface ai.djl.ndarray.BytesSupplier
      • set

        public void set​(java.nio.Buffer buffer)
        Specified by:
        set in interface ai.djl.ndarray.NDArray
      • gather

        public ai.djl.ndarray.NDArray gather​(ai.djl.ndarray.NDArray index,
                                             int axis)
        Specified by:
        gather in interface ai.djl.ndarray.NDArray
      • gatherNd

        public ai.djl.ndarray.NDArray gatherNd​(ai.djl.ndarray.NDArray index)
        Specified by:
        gatherNd in interface ai.djl.ndarray.NDArray
      • take

        public ai.djl.ndarray.NDArray take​(ai.djl.ndarray.NDManager manager,
                                           ai.djl.ndarray.NDArray index)
        Specified by:
        take in interface ai.djl.ndarray.NDArray
      • put

        public ai.djl.ndarray.NDArray put​(ai.djl.ndarray.NDArray index,
                                          ai.djl.ndarray.NDArray value)
        Specified by:
        put in interface ai.djl.ndarray.NDArray
      • scatter

        public ai.djl.ndarray.NDArray scatter​(ai.djl.ndarray.NDArray index,
                                              ai.djl.ndarray.NDArray value,
                                              int axis)
        Specified by:
        scatter in interface ai.djl.ndarray.NDArray
      • attach

        public void attach​(ai.djl.ndarray.NDManager manager)
        Specified by:
        attach in interface ai.djl.ndarray.NDResource
      • returnResource

        public void returnResource​(ai.djl.ndarray.NDManager manager)
        Specified by:
        returnResource in interface ai.djl.ndarray.NDResource
      • tempAttach

        public void tempAttach​(ai.djl.ndarray.NDManager manager)
        Specified by:
        tempAttach in interface ai.djl.ndarray.NDResource
      • detach

        public void detach()
        Specified by:
        detach in interface ai.djl.ndarray.NDResource
      • duplicate

        public ai.djl.ndarray.NDArray duplicate()
        Specified by:
        duplicate in interface ai.djl.ndarray.NDArray
      • booleanMask

        public ai.djl.ndarray.NDArray booleanMask​(ai.djl.ndarray.NDArray index,
                                                  int axis)
        Specified by:
        booleanMask in interface ai.djl.ndarray.NDArray
      • sequenceMask

        public ai.djl.ndarray.NDArray sequenceMask​(ai.djl.ndarray.NDArray sequenceLength,
                                                   float value)
        Specified by:
        sequenceMask in interface ai.djl.ndarray.NDArray
      • sequenceMask

        public ai.djl.ndarray.NDArray sequenceMask​(ai.djl.ndarray.NDArray sequenceLength)
        Specified by:
        sequenceMask in interface ai.djl.ndarray.NDArray
      • contentEquals

        public boolean contentEquals​(java.lang.Number number)
        Specified by:
        contentEquals in interface ai.djl.ndarray.NDArray
      • contentEquals

        public boolean contentEquals​(ai.djl.ndarray.NDArray other)
        Specified by:
        contentEquals in interface ai.djl.ndarray.NDArray
      • eq

        public ai.djl.ndarray.NDArray eq​(java.lang.Number n)
        Specified by:
        eq in interface ai.djl.ndarray.NDArray
      • eq

        public ai.djl.ndarray.NDArray eq​(ai.djl.ndarray.NDArray other)
        Specified by:
        eq in interface ai.djl.ndarray.NDArray
      • neq

        public ai.djl.ndarray.NDArray neq​(java.lang.Number n)
        Specified by:
        neq in interface ai.djl.ndarray.NDArray
      • neq

        public ai.djl.ndarray.NDArray neq​(ai.djl.ndarray.NDArray other)
        Specified by:
        neq in interface ai.djl.ndarray.NDArray
      • gt

        public ai.djl.ndarray.NDArray gt​(java.lang.Number n)
        Specified by:
        gt in interface ai.djl.ndarray.NDArray
      • gt

        public ai.djl.ndarray.NDArray gt​(ai.djl.ndarray.NDArray other)
        Specified by:
        gt in interface ai.djl.ndarray.NDArray
      • gte

        public ai.djl.ndarray.NDArray gte​(java.lang.Number n)
        Specified by:
        gte in interface ai.djl.ndarray.NDArray
      • gte

        public ai.djl.ndarray.NDArray gte​(ai.djl.ndarray.NDArray other)
        Specified by:
        gte in interface ai.djl.ndarray.NDArray
      • lt

        public ai.djl.ndarray.NDArray lt​(java.lang.Number n)
        Specified by:
        lt in interface ai.djl.ndarray.NDArray
      • lt

        public ai.djl.ndarray.NDArray lt​(ai.djl.ndarray.NDArray other)
        Specified by:
        lt in interface ai.djl.ndarray.NDArray
      • lte

        public ai.djl.ndarray.NDArray lte​(java.lang.Number n)
        Specified by:
        lte in interface ai.djl.ndarray.NDArray
      • lte

        public ai.djl.ndarray.NDArray lte​(ai.djl.ndarray.NDArray other)
        Specified by:
        lte in interface ai.djl.ndarray.NDArray
      • all

        public ai.djl.ndarray.NDArray all()
        Specified by:
        all in interface ai.djl.ndarray.NDArray
      • any

        public ai.djl.ndarray.NDArray any()
        Specified by:
        any in interface ai.djl.ndarray.NDArray
      • erfinv

        public ai.djl.ndarray.NDArray erfinv()
        Specified by:
        erfinv in interface ai.djl.ndarray.NDArray
      • norm

        public ai.djl.ndarray.NDArray norm​(boolean keepDims)
        Specified by:
        norm in interface ai.djl.ndarray.NDArray
      • norm

        public ai.djl.ndarray.NDArray norm​(int ord,
                                           int[] axes,
                                           boolean keepDims)
        Specified by:
        norm in interface ai.djl.ndarray.NDArray
      • oneHot

        public ai.djl.ndarray.NDArray oneHot​(int depth,
                                             float onValue,
                                             float offValue,
                                             ai.djl.ndarray.types.DataType dataType)
        Specified by:
        oneHot in interface ai.djl.ndarray.NDArray
      • batchDot

        public ai.djl.ndarray.NDArray batchDot​(ai.djl.ndarray.NDArray other)
        Specified by:
        batchDot in interface ai.djl.ndarray.NDArray
      • complex

        public ai.djl.ndarray.NDArray complex()
        Specified by:
        complex in interface ai.djl.ndarray.NDArray
      • real

        public ai.djl.ndarray.NDArray real()
        Specified by:
        real in interface ai.djl.ndarray.NDArray
      • add

        public ai.djl.ndarray.NDArray add​(java.lang.Number n)
        Specified by:
        add in interface ai.djl.ndarray.NDArray
      • add

        public ai.djl.ndarray.NDArray add​(ai.djl.ndarray.NDArray other)
        Specified by:
        add in interface ai.djl.ndarray.NDArray
      • sub

        public ai.djl.ndarray.NDArray sub​(java.lang.Number n)
        Specified by:
        sub in interface ai.djl.ndarray.NDArray
      • sub

        public ai.djl.ndarray.NDArray sub​(ai.djl.ndarray.NDArray other)
        Specified by:
        sub in interface ai.djl.ndarray.NDArray
      • mul

        public ai.djl.ndarray.NDArray mul​(java.lang.Number n)
        Specified by:
        mul in interface ai.djl.ndarray.NDArray
      • mul

        public ai.djl.ndarray.NDArray mul​(ai.djl.ndarray.NDArray other)
        Specified by:
        mul in interface ai.djl.ndarray.NDArray
      • div

        public ai.djl.ndarray.NDArray div​(java.lang.Number n)
        Specified by:
        div in interface ai.djl.ndarray.NDArray
      • div

        public ai.djl.ndarray.NDArray div​(ai.djl.ndarray.NDArray other)
        Specified by:
        div in interface ai.djl.ndarray.NDArray
      • mod

        public ai.djl.ndarray.NDArray mod​(java.lang.Number n)
        Specified by:
        mod in interface ai.djl.ndarray.NDArray
      • mod

        public ai.djl.ndarray.NDArray mod​(ai.djl.ndarray.NDArray other)
        Specified by:
        mod in interface ai.djl.ndarray.NDArray
      • pow

        public ai.djl.ndarray.NDArray pow​(java.lang.Number n)
        Specified by:
        pow in interface ai.djl.ndarray.NDArray
      • pow

        public ai.djl.ndarray.NDArray pow​(ai.djl.ndarray.NDArray other)
        Specified by:
        pow in interface ai.djl.ndarray.NDArray
      • maximum

        public ai.djl.ndarray.NDArray maximum​(java.lang.Number n)
        Specified by:
        maximum in interface ai.djl.ndarray.NDArray
      • maximum

        public ai.djl.ndarray.NDArray maximum​(ai.djl.ndarray.NDArray other)
        Specified by:
        maximum in interface ai.djl.ndarray.NDArray
      • minimum

        public ai.djl.ndarray.NDArray minimum​(java.lang.Number n)
        Specified by:
        minimum in interface ai.djl.ndarray.NDArray
      • minimum

        public ai.djl.ndarray.NDArray minimum​(ai.djl.ndarray.NDArray other)
        Specified by:
        minimum in interface ai.djl.ndarray.NDArray
      • addi

        public ai.djl.ndarray.NDArray addi​(java.lang.Number n)
        Specified by:
        addi in interface ai.djl.ndarray.NDArray
      • addi

        public ai.djl.ndarray.NDArray addi​(ai.djl.ndarray.NDArray other)
        Specified by:
        addi in interface ai.djl.ndarray.NDArray
      • subi

        public ai.djl.ndarray.NDArray subi​(java.lang.Number n)
        Specified by:
        subi in interface ai.djl.ndarray.NDArray
      • subi

        public ai.djl.ndarray.NDArray subi​(ai.djl.ndarray.NDArray other)
        Specified by:
        subi in interface ai.djl.ndarray.NDArray
      • muli

        public ai.djl.ndarray.NDArray muli​(java.lang.Number n)
        Specified by:
        muli in interface ai.djl.ndarray.NDArray
      • muli

        public ai.djl.ndarray.NDArray muli​(ai.djl.ndarray.NDArray other)
        Specified by:
        muli in interface ai.djl.ndarray.NDArray
      • divi

        public ai.djl.ndarray.NDArray divi​(java.lang.Number n)
        Specified by:
        divi in interface ai.djl.ndarray.NDArray
      • divi

        public ai.djl.ndarray.NDArray divi​(ai.djl.ndarray.NDArray other)
        Specified by:
        divi in interface ai.djl.ndarray.NDArray
      • toSparse

        public ai.djl.ndarray.NDArray toSparse​(ai.djl.ndarray.types.SparseFormat fmt)
        Specified by:
        toSparse in interface ai.djl.ndarray.NDArray
      • modi

        public ai.djl.ndarray.NDArray modi​(java.lang.Number n)
        Specified by:
        modi in interface ai.djl.ndarray.NDArray
      • modi

        public ai.djl.ndarray.NDArray modi​(ai.djl.ndarray.NDArray other)
        Specified by:
        modi in interface ai.djl.ndarray.NDArray
      • powi

        public ai.djl.ndarray.NDArray powi​(java.lang.Number n)
        Specified by:
        powi in interface ai.djl.ndarray.NDArray
      • powi

        public ai.djl.ndarray.NDArray powi​(ai.djl.ndarray.NDArray other)
        Specified by:
        powi in interface ai.djl.ndarray.NDArray
      • sign

        public ai.djl.ndarray.NDArray sign()
        Specified by:
        sign in interface ai.djl.ndarray.NDArray
      • signi

        public ai.djl.ndarray.NDArray signi()
        Specified by:
        signi in interface ai.djl.ndarray.NDArray
      • neg

        public ai.djl.ndarray.NDArray neg()
        Specified by:
        neg in interface ai.djl.ndarray.NDArray
      • negi

        public ai.djl.ndarray.NDArray negi()
        Specified by:
        negi in interface ai.djl.ndarray.NDArray
      • abs

        public ai.djl.ndarray.NDArray abs()
        Specified by:
        abs in interface ai.djl.ndarray.NDArray
      • square

        public ai.djl.ndarray.NDArray square()
        Specified by:
        square in interface ai.djl.ndarray.NDArray
      • sqrt

        public ai.djl.ndarray.NDArray sqrt()
        Specified by:
        sqrt in interface ai.djl.ndarray.NDArray
      • cbrt

        public ai.djl.ndarray.NDArray cbrt()
        Specified by:
        cbrt in interface ai.djl.ndarray.NDArray
      • floor

        public ai.djl.ndarray.NDArray floor()
        Specified by:
        floor in interface ai.djl.ndarray.NDArray
      • ceil

        public ai.djl.ndarray.NDArray ceil()
        Specified by:
        ceil in interface ai.djl.ndarray.NDArray
      • round

        public ai.djl.ndarray.NDArray round()
        Specified by:
        round in interface ai.djl.ndarray.NDArray
      • trunc

        public ai.djl.ndarray.NDArray trunc()
        Specified by:
        trunc in interface ai.djl.ndarray.NDArray
      • exp

        public ai.djl.ndarray.NDArray exp()
        Specified by:
        exp in interface ai.djl.ndarray.NDArray
      • gammaln

        public ai.djl.ndarray.NDArray gammaln()
        Specified by:
        gammaln in interface ai.djl.ndarray.NDArray
      • log

        public ai.djl.ndarray.NDArray log()
        Specified by:
        log in interface ai.djl.ndarray.NDArray
      • log10

        public ai.djl.ndarray.NDArray log10()
        Specified by:
        log10 in interface ai.djl.ndarray.NDArray
      • log2

        public ai.djl.ndarray.NDArray log2()
        Specified by:
        log2 in interface ai.djl.ndarray.NDArray
      • sin

        public ai.djl.ndarray.NDArray sin()
        Specified by:
        sin in interface ai.djl.ndarray.NDArray
      • cos

        public ai.djl.ndarray.NDArray cos()
        Specified by:
        cos in interface ai.djl.ndarray.NDArray
      • tan

        public ai.djl.ndarray.NDArray tan()
        Specified by:
        tan in interface ai.djl.ndarray.NDArray
      • asin

        public ai.djl.ndarray.NDArray asin()
        Specified by:
        asin in interface ai.djl.ndarray.NDArray
      • acos

        public ai.djl.ndarray.NDArray acos()
        Specified by:
        acos in interface ai.djl.ndarray.NDArray
      • atan

        public ai.djl.ndarray.NDArray atan()
        Specified by:
        atan in interface ai.djl.ndarray.NDArray
      • sinh

        public ai.djl.ndarray.NDArray sinh()
        Specified by:
        sinh in interface ai.djl.ndarray.NDArray
      • cosh

        public ai.djl.ndarray.NDArray cosh()
        Specified by:
        cosh in interface ai.djl.ndarray.NDArray
      • tanh

        public ai.djl.ndarray.NDArray tanh()
        Specified by:
        tanh in interface ai.djl.ndarray.NDArray
      • asinh

        public ai.djl.ndarray.NDArray asinh()
        Specified by:
        asinh in interface ai.djl.ndarray.NDArray
      • acosh

        public ai.djl.ndarray.NDArray acosh()
        Specified by:
        acosh in interface ai.djl.ndarray.NDArray
      • atanh

        public ai.djl.ndarray.NDArray atanh()
        Specified by:
        atanh in interface ai.djl.ndarray.NDArray
      • toDegrees

        public ai.djl.ndarray.NDArray toDegrees()
        Specified by:
        toDegrees in interface ai.djl.ndarray.NDArray
      • toRadians

        public ai.djl.ndarray.NDArray toRadians()
        Specified by:
        toRadians in interface ai.djl.ndarray.NDArray
      • max

        public ai.djl.ndarray.NDArray max()
        Specified by:
        max in interface ai.djl.ndarray.NDArray
      • max

        public ai.djl.ndarray.NDArray max​(int[] axes,
                                          boolean keepDims)
        Specified by:
        max in interface ai.djl.ndarray.NDArray
      • min

        public ai.djl.ndarray.NDArray min()
        Specified by:
        min in interface ai.djl.ndarray.NDArray
      • min

        public ai.djl.ndarray.NDArray min​(int[] axes,
                                          boolean keepDims)
        Specified by:
        min in interface ai.djl.ndarray.NDArray
      • sum

        public ai.djl.ndarray.NDArray sum()
        Specified by:
        sum in interface ai.djl.ndarray.NDArray
      • sum

        public ai.djl.ndarray.NDArray sum​(int[] axes,
                                          boolean keepDims)
        Specified by:
        sum in interface ai.djl.ndarray.NDArray
      • cumProd

        public ai.djl.ndarray.NDArray cumProd​(int axis)
        Specified by:
        cumProd in interface ai.djl.ndarray.NDArray
      • cumProd

        public ai.djl.ndarray.NDArray cumProd​(int axis,
                                              ai.djl.ndarray.types.DataType dataType)
        Specified by:
        cumProd in interface ai.djl.ndarray.NDArray
      • prod

        public ai.djl.ndarray.NDArray prod()
        Specified by:
        prod in interface ai.djl.ndarray.NDArray
      • prod

        public ai.djl.ndarray.NDArray prod​(int[] axes,
                                           boolean keepDims)
        Specified by:
        prod in interface ai.djl.ndarray.NDArray
      • mean

        public ai.djl.ndarray.NDArray mean()
        Specified by:
        mean in interface ai.djl.ndarray.NDArray
      • mean

        public ai.djl.ndarray.NDArray mean​(int[] axes,
                                           boolean keepDims)
        Specified by:
        mean in interface ai.djl.ndarray.NDArray
      • normalize

        public ai.djl.ndarray.NDArray normalize​(double p,
                                                long dim,
                                                double eps)
        Specified by:
        normalize in interface ai.djl.ndarray.NDArray
      • rotate90

        public ai.djl.ndarray.NDArray rotate90​(int times,
                                               int[] axes)
        Specified by:
        rotate90 in interface ai.djl.ndarray.NDArray
      • trace

        public ai.djl.ndarray.NDArray trace​(int offset,
                                            int axis1,
                                            int axis2)
        Specified by:
        trace in interface ai.djl.ndarray.NDArray
      • split

        public ai.djl.ndarray.NDList split​(long[] indices,
                                           int axis)
        Specified by:
        split in interface ai.djl.ndarray.NDArray
      • flatten

        public ai.djl.ndarray.NDArray flatten()
        Specified by:
        flatten in interface ai.djl.ndarray.NDArray
      • flatten

        public ai.djl.ndarray.NDArray flatten​(int startDim,
                                              int endDim)
        Specified by:
        flatten in interface ai.djl.ndarray.NDArray
      • fft

        public ai.djl.ndarray.NDArray fft​(long length,
                                          long axis)
        Specified by:
        fft in interface ai.djl.ndarray.NDArray
      • stft

        public ai.djl.ndarray.NDArray stft​(long nFft,
                                           long hopLength,
                                           boolean center,
                                           ai.djl.ndarray.NDArray window,
                                           boolean normalize,
                                           boolean returnComplex)
        Specified by:
        stft in interface ai.djl.ndarray.NDArray
      • reshape

        public ai.djl.ndarray.NDArray reshape​(ai.djl.ndarray.types.Shape shape)
        Specified by:
        reshape in interface ai.djl.ndarray.NDArray
      • expandDims

        public ai.djl.ndarray.NDArray expandDims​(int axis)
        Specified by:
        expandDims in interface ai.djl.ndarray.NDArray
      • squeeze

        public ai.djl.ndarray.NDArray squeeze​(int[] axes)
        Specified by:
        squeeze in interface ai.djl.ndarray.NDArray
      • unique

        public ai.djl.ndarray.NDList unique​(java.lang.Integer dim,
                                            boolean sorted,
                                            boolean returnInverse,
                                            boolean returnCounts)
        Specified by:
        unique in interface ai.djl.ndarray.NDArray
      • logicalAnd

        public ai.djl.ndarray.NDArray logicalAnd​(ai.djl.ndarray.NDArray n)
        Specified by:
        logicalAnd in interface ai.djl.ndarray.NDArray
      • logicalOr

        public ai.djl.ndarray.NDArray logicalOr​(ai.djl.ndarray.NDArray n)
        Specified by:
        logicalOr in interface ai.djl.ndarray.NDArray
      • logicalXor

        public ai.djl.ndarray.NDArray logicalXor​(ai.djl.ndarray.NDArray n)
        Specified by:
        logicalXor in interface ai.djl.ndarray.NDArray
      • logicalNot

        public ai.djl.ndarray.NDArray logicalNot()
        Specified by:
        logicalNot in interface ai.djl.ndarray.NDArray
      • argSort

        public ai.djl.ndarray.NDArray argSort​(int axis,
                                              boolean ascending)
        Specified by:
        argSort in interface ai.djl.ndarray.NDArray
      • sort

        public ai.djl.ndarray.NDArray sort​(int axis)
        Specified by:
        sort in interface ai.djl.ndarray.NDArray
      • sort

        public ai.djl.ndarray.NDArray sort()
        Specified by:
        sort in interface ai.djl.ndarray.NDArray
      • softmax

        public ai.djl.ndarray.NDArray softmax​(int axis)
        Specified by:
        softmax in interface ai.djl.ndarray.NDArray
      • logSoftmax

        public ai.djl.ndarray.NDArray logSoftmax​(int axis)
        Specified by:
        logSoftmax in interface ai.djl.ndarray.NDArray
      • cumSum

        public ai.djl.ndarray.NDArray cumSum​(int axis)
        Specified by:
        cumSum in interface ai.djl.ndarray.NDArray
      • cumSum

        public ai.djl.ndarray.NDArray cumSum()
        Specified by:
        cumSum in interface ai.djl.ndarray.NDArray
      • intern

        public void intern​(ai.djl.ndarray.NDArray replaced)
        Specified by:
        intern in interface ai.djl.ndarray.NDArray
      • isInfinite

        public ai.djl.ndarray.NDArray isInfinite()
        Specified by:
        isInfinite in interface ai.djl.ndarray.NDArray
      • inverse

        public ai.djl.ndarray.NDArray inverse()
        Specified by:
        inverse in interface ai.djl.ndarray.NDArray
      • isNaN

        public ai.djl.ndarray.NDArray isNaN()
        Specified by:
        isNaN in interface ai.djl.ndarray.NDArray
      • tile

        public ai.djl.ndarray.NDArray tile​(long repeats)
        Specified by:
        tile in interface ai.djl.ndarray.NDArray
      • tile

        public ai.djl.ndarray.NDArray tile​(int axis,
                                           long repeats)
        Specified by:
        tile in interface ai.djl.ndarray.NDArray
      • tile

        public ai.djl.ndarray.NDArray tile​(long[] repeats)
        Specified by:
        tile in interface ai.djl.ndarray.NDArray
      • tile

        public ai.djl.ndarray.NDArray tile​(ai.djl.ndarray.types.Shape desiredShape)
        Specified by:
        tile in interface ai.djl.ndarray.NDArray
      • repeat

        public ai.djl.ndarray.NDArray repeat​(long repeats)
        Specified by:
        repeat in interface ai.djl.ndarray.NDArray
      • repeat

        public ai.djl.ndarray.NDArray repeat​(int axis,
                                             long repeats)
        Specified by:
        repeat in interface ai.djl.ndarray.NDArray
      • repeat

        public ai.djl.ndarray.NDArray repeat​(long[] repeats)
        Specified by:
        repeat in interface ai.djl.ndarray.NDArray
      • repeat

        public ai.djl.ndarray.NDArray repeat​(ai.djl.ndarray.types.Shape desiredShape)
        Specified by:
        repeat in interface ai.djl.ndarray.NDArray
      • dot

        public ai.djl.ndarray.NDArray dot​(ai.djl.ndarray.NDArray other)
        Specified by:
        dot in interface ai.djl.ndarray.NDArray
      • matMul

        public ai.djl.ndarray.NDArray matMul​(ai.djl.ndarray.NDArray other)
        Specified by:
        matMul in interface ai.djl.ndarray.NDArray
      • batchMatMul

        public ai.djl.ndarray.NDArray batchMatMul​(ai.djl.ndarray.NDArray other)
        Specified by:
        batchMatMul in interface ai.djl.ndarray.NDArray
      • xlogy

        public ai.djl.ndarray.NDArray xlogy​(ai.djl.ndarray.NDArray other)
        Specified by:
        xlogy in interface ai.djl.ndarray.NDArray
      • clip

        public ai.djl.ndarray.NDArray clip​(java.lang.Number min,
                                           java.lang.Number max)
        Specified by:
        clip in interface ai.djl.ndarray.NDArray
      • flip

        public ai.djl.ndarray.NDArray flip​(int... axes)
        Specified by:
        flip in interface ai.djl.ndarray.NDArray
      • transpose

        public ai.djl.ndarray.NDArray transpose()
        Specified by:
        transpose in interface ai.djl.ndarray.NDArray
      • transpose

        public ai.djl.ndarray.NDArray transpose​(int... dimensions)
        Specified by:
        transpose in interface ai.djl.ndarray.NDArray
      • broadcast

        public ai.djl.ndarray.NDArray broadcast​(ai.djl.ndarray.types.Shape shape)
        Specified by:
        broadcast in interface ai.djl.ndarray.NDArray
      • argMax

        public ai.djl.ndarray.NDArray argMax()
        Specified by:
        argMax in interface ai.djl.ndarray.NDArray
      • argMax

        public ai.djl.ndarray.NDArray argMax​(int axis)
        Specified by:
        argMax in interface ai.djl.ndarray.NDArray
      • argMin

        public ai.djl.ndarray.NDArray argMin()
        Specified by:
        argMin in interface ai.djl.ndarray.NDArray
      • argMin

        public ai.djl.ndarray.NDArray argMin​(int axis)
        Specified by:
        argMin in interface ai.djl.ndarray.NDArray
      • percentile

        public ai.djl.ndarray.NDArray percentile​(java.lang.Number percentile)
        Specified by:
        percentile in interface ai.djl.ndarray.NDArray
      • percentile

        public ai.djl.ndarray.NDArray percentile​(java.lang.Number percentile,
                                                 int[] dimension)
        Specified by:
        percentile in interface ai.djl.ndarray.NDArray
      • median

        public ai.djl.ndarray.NDArray median()
        Specified by:
        median in interface ai.djl.ndarray.NDArray
      • median

        public ai.djl.ndarray.NDArray median​(int[] axes)
        Specified by:
        median in interface ai.djl.ndarray.NDArray
      • toDense

        public ai.djl.ndarray.NDArray toDense()
        Specified by:
        toDense in interface ai.djl.ndarray.NDArray
      • nonzero

        public ai.djl.ndarray.NDArray nonzero()
        Specified by:
        nonzero in interface ai.djl.ndarray.NDArray
      • getNDArrayInternal

        public ai.djl.ndarray.internal.NDArrayEx getNDArrayInternal()
        Specified by:
        getNDArrayInternal in interface ai.djl.ndarray.NDArray
      • equals

        public boolean equals​(java.lang.Object obj)
        Overrides:
        equals in class java.lang.Object
      • hashCode

        public int hashCode()
        Overrides:
        hashCode in class java.lang.Object
      • toString

        public java.lang.String toString()
        Overrides:
        toString in class java.lang.Object
      • close

        public void close()
        Specified by:
        close in interface java.lang.AutoCloseable
        Specified by:
        close in interface ai.djl.ndarray.NDArray
        Specified by:
        close in interface ai.djl.ndarray.NDResource
        Overrides:
        close in class ai.djl.util.NativeResource<org.tensorflow.internal.c_api.TFE_TensorHandle>
      • setHandle

        protected void setHandle​(org.tensorflow.internal.c_api.TFE_TensorHandle newHandle)