Package ai.djl.pytorch.engine

Class PtNDArray

java.lang.Object
ai.djl.util.NativeResource<Long>
ai.djl.pytorch.engine.PtNDArray
All Implemented Interfaces:
ai.djl.ndarray.BytesSupplier, ai.djl.ndarray.NDArray, ai.djl.ndarray.NDResource, AutoCloseable
public class PtNDArray extends ai.djl.util.NativeResource<Long> implements ai.djl.ndarray.NDArray
PtNDArray is the PyTorch implementation of NDArray.

  • Constructor Details

    • 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, 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, 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 Details

    • getManager

      public PtNDManager getManager()
      Specified by:
      getManager in interface ai.djl.ndarray.NDResource

    • getName

      public String getName()
      Specified by:
      getName in interface ai.djl.ndarray.NDArray

    • setName

      public void setName(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 ByteBuffer toByteBuffer(boolean tryDirect)
      Specified by:
      toByteBuffer in interface ai.djl.ndarray.NDArray

    • toStringArray

      public String[] toStringArray(Charset charset)
      Specified by:
      toStringArray in interface ai.djl.ndarray.NDArray

    • set

      public void set(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(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(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(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(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(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(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(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(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(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(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(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(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(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(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(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(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(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(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(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(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(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

    • atan2

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

    • rfft

      public ai.djl.ndarray.NDArray rfft(long length, long axis)
      Specified by:
      rfft in interface ai.djl.ndarray.NDArray

    • ifft

      public ai.djl.ndarray.NDArray ifft(long length, long axis)
      Specified by:
      ifft in interface ai.djl.ndarray.NDArray

    • irfft

      public ai.djl.ndarray.NDArray irfft(long length, long axis)
      Specified by:
      irfft 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

    • fft2

      public ai.djl.ndarray.NDArray fft2(long[] sizes, long[] axes)
      Specified by:
      fft2 in interface ai.djl.ndarray.NDArray

    • ifft2

      public ai.djl.ndarray.NDArray ifft2(long[] sizes, long[] axes)
      Specified by:
      ifft2 in interface ai.djl.ndarray.NDArray

    • pad

      public ai.djl.ndarray.NDArray pad(ai.djl.ndarray.types.Shape padding, double value)
      Specified by:
      pad 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(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

    • diagonal

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

    • diagonal

      public ai.djl.ndarray.NDArray diagonal(int offset)
      Specified by:
      diagonal in interface ai.djl.ndarray.NDArray

    • diagonal

      public ai.djl.ndarray.NDArray diagonal(int offset, int axis1, int axis2)
      Specified by:
      diagonal 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(Number min, 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(Number percentile)
      Specified by:
      percentile in interface ai.djl.ndarray.NDArray

    • percentile

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

    • erf

      public PtNDArray erf()
      Specified by:
      erf 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

    • conj

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

    • diff

      public ai.djl.ndarray.NDArray diff(int n, int dim)
      Specified by:
      diff in interface ai.djl.ndarray.NDArray

    • getNDArrayInternal

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

    • toString

      public String toString()
      Overrides:
      toString in class Object

    • equals

      public boolean equals(Object obj)
      Overrides:
      equals in class Object

    • hashCode

      public int hashCode()
      Overrides:
      hashCode in class Object

    • close

      public void close()
      Specified by:
      close in interface 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<Long>