Class PtNDArray

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

    public class PtNDArray
    extends ai.djl.util.NativeResource<java.lang.Long>
    implements ai.djl.ndarray.NDArray
    PtNDArray is the PyTorch implementation of NDArray.
    • Constructor Detail

      • PtNDArray

        public PtNDArray​(PtNDManager manager,
                         long handle)
        Constructs a PyTorch NDArray from a native handle (internal. Use NDManager instead).
        Parameters:
        manager - the manager to attach the new array to
        handle - the pointer to the native PyTorch memory
      • PtNDArray

        public PtNDArray​(PtNDManager manager,
                         long handle,
                         java.nio.ByteBuffer data)
        Constructs a PyTorch NDArray from a native handle (internal. Use NDManager instead) with the data that is hold on Java side.
        Parameters:
        manager - the manager to attach the new array to
        handle - the pointer to the native PyTorch memory
        data - the direct buffer of the data
      • PtNDArray

        public PtNDArray​(PtNDManager manager,
                         java.lang.String[] strs,
                         ai.djl.ndarray.types.Shape shape)
        Constructs a PyTorch NDArray to hold string array with a dummy native handle (internal. Use NDManager instead) with the data that is hold on Java side.
        Parameters:
        manager - the manager to attach the new array to
        strs - the string array
        shape - the Shape of the NDArray
    • Method Detail

      • getManager

        public PtNDManager 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 PtNDArray toDevice​(ai.djl.Device device,
                                  boolean copy)
        Specified by:
        toDevice in interface ai.djl.ndarray.NDArray
      • toType

        public PtNDArray 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 PtNDArray 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
      • toByteBuffer

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

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

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

        public ai.djl.ndarray.NDArray get​(ai.djl.ndarray.NDManager manager,
                                          long... indices)
        Specified by:
        get 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 PtNDArray 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 PtNDArray eq​(java.lang.Number n)
        Specified by:
        eq in interface ai.djl.ndarray.NDArray
      • eq

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

        public PtNDArray pow​(ai.djl.ndarray.NDArray other)
        Specified by:
        pow 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
      • addi

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

        public PtNDArray 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 PtNDArray cbrt()
        Specified by:
        cbrt in interface ai.djl.ndarray.NDArray
      • floor

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

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

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

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

        public PtNDArray 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 PtNDArray log()
        Specified by:
        log in interface ai.djl.ndarray.NDArray
      • log10

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

        public PtNDArray 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 PtNDArray prod()
        Specified by:
        prod in interface ai.djl.ndarray.NDArray
      • prod

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

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

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

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

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

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

        public ai.djl.ndarray.NDList split​(long sections,
                                           int axis)
        Specified by:
        split 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 PtNDArray 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 PtNDArray reshape​(ai.djl.ndarray.types.Shape shape)
        Specified by:
        reshape in interface ai.djl.ndarray.NDArray
      • expandDims

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

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

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

        public PtNDArray 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 PtNDArray logicalAnd​(ai.djl.ndarray.NDArray other)
        Specified by:
        logicalAnd in interface ai.djl.ndarray.NDArray
      • logicalOr

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

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

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

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

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

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

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

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

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

        public PtNDArray cumSum​(int axis)
        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 PtNDArray isInfinite()
        Specified by:
        isInfinite in interface ai.djl.ndarray.NDArray
      • isNaN

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

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

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

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

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

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

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

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

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

        public PtNDArray 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
      • clip

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

        public PtNDArray swapAxes​(int axis1,
                                  int axis2)
        Specified by:
        swapAxes 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 PtNDArray transpose()
        Specified by:
        transpose in interface ai.djl.ndarray.NDArray
      • transpose

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

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

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

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

        public ai.djl.ndarray.NDList topK​(int k,
                                          int axis,
                                          boolean largest,
                                          boolean sorted)
        Specified by:
        topK in interface ai.djl.ndarray.NDArray
      • argMin

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

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

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

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

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

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

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

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

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

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

        public PtNDArray inverse()
        Specified by:
        inverse 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 order,
                                           int[] axes,
                                           boolean keepDims)
        Specified by:
        norm in interface ai.djl.ndarray.NDArray
      • oneHot

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

        public ai.djl.ndarray.NDArray oneHot​(int depth,
                                             ai.djl.ndarray.types.DataType dataType)
        Specified by:
        oneHot 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
      • getNDArrayInternal

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

        public java.lang.String toString()
        Overrides:
        toString in class java.lang.Object
      • 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
      • 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<java.lang.Long>