Package ai.djl.engine.rust

Class RsNDArray

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

  • Constructor Details

    • RsNDArray

      public RsNDArray(RsNDManager manager, long handle)
      Constructs a Rust 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 Rust memory

    • RsNDArray

      public RsNDArray(RsNDManager manager, long handle, ai.djl.ndarray.types.DataType dataType, ByteBuffer data)
      Constructs a Rust 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 Rust memory
      dataType - the DataType to be set
      data - the direct buffer of the data

  • Method Details

    • getManager

      public RsNDManager 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 RsNDArray toDevice(ai.djl.Device device, boolean copy)
      Specified by:
      toDevice in interface ai.djl.ndarray.NDArray

    • toType

      public RsNDArray 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 RsNDArray 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

    • 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 RsNDArray 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 RsNDArray eq(Number n)
      Specified by:
      eq in interface ai.djl.ndarray.NDArray

    • eq

      public RsNDArray eq(ai.djl.ndarray.NDArray other)
      Specified by:
      eq in interface ai.djl.ndarray.NDArray

    • neq

      public RsNDArray neq(Number n)
      Specified by:
      neq in interface ai.djl.ndarray.NDArray

    • neq

      public RsNDArray neq(ai.djl.ndarray.NDArray other)
      Specified by:
      neq in interface ai.djl.ndarray.NDArray

    • gt

      public RsNDArray gt(Number n)
      Specified by:
      gt in interface ai.djl.ndarray.NDArray

    • gt

      public RsNDArray gt(ai.djl.ndarray.NDArray other)
      Specified by:
      gt in interface ai.djl.ndarray.NDArray

    • gte

      public RsNDArray gte(Number n)
      Specified by:
      gte in interface ai.djl.ndarray.NDArray

    • gte

      public RsNDArray gte(ai.djl.ndarray.NDArray other)
      Specified by:
      gte in interface ai.djl.ndarray.NDArray

    • lt

      public RsNDArray lt(Number n)
      Specified by:
      lt in interface ai.djl.ndarray.NDArray

    • lt

      public RsNDArray lt(ai.djl.ndarray.NDArray other)
      Specified by:
      lt in interface ai.djl.ndarray.NDArray

    • lte

      public RsNDArray lte(Number n)
      Specified by:
      lte in interface ai.djl.ndarray.NDArray

    • lte

      public RsNDArray lte(ai.djl.ndarray.NDArray other)
      Specified by:
      lte in interface ai.djl.ndarray.NDArray

    • add

      public RsNDArray add(Number n)
      Specified by:
      add in interface ai.djl.ndarray.NDArray

    • add

      public RsNDArray add(ai.djl.ndarray.NDArray other)
      Specified by:
      add in interface ai.djl.ndarray.NDArray

    • sub

      public RsNDArray sub(Number n)
      Specified by:
      sub in interface ai.djl.ndarray.NDArray

    • sub

      public RsNDArray sub(ai.djl.ndarray.NDArray other)
      Specified by:
      sub in interface ai.djl.ndarray.NDArray

    • mul

      public RsNDArray mul(Number n)
      Specified by:
      mul in interface ai.djl.ndarray.NDArray

    • mul

      public RsNDArray mul(ai.djl.ndarray.NDArray other)
      Specified by:
      mul in interface ai.djl.ndarray.NDArray

    • div

      public RsNDArray div(Number n)
      Specified by:
      div in interface ai.djl.ndarray.NDArray

    • div

      public RsNDArray div(ai.djl.ndarray.NDArray other)
      Specified by:
      div in interface ai.djl.ndarray.NDArray

    • mod

      public RsNDArray mod(Number n)
      Specified by:
      mod in interface ai.djl.ndarray.NDArray

    • mod

      public RsNDArray mod(ai.djl.ndarray.NDArray other)
      Specified by:
      mod in interface ai.djl.ndarray.NDArray

    • pow

      public RsNDArray pow(Number n)
      Specified by:
      pow in interface ai.djl.ndarray.NDArray

    • pow

      public RsNDArray 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 RsNDArray addi(Number n)
      Specified by:
      addi in interface ai.djl.ndarray.NDArray

    • addi

      public RsNDArray addi(ai.djl.ndarray.NDArray other)
      Specified by:
      addi in interface ai.djl.ndarray.NDArray

    • subi

      public RsNDArray subi(Number n)
      Specified by:
      subi in interface ai.djl.ndarray.NDArray

    • subi

      public RsNDArray subi(ai.djl.ndarray.NDArray other)
      Specified by:
      subi in interface ai.djl.ndarray.NDArray

    • muli

      public RsNDArray muli(Number n)
      Specified by:
      muli in interface ai.djl.ndarray.NDArray

    • muli

      public RsNDArray muli(ai.djl.ndarray.NDArray other)
      Specified by:
      muli in interface ai.djl.ndarray.NDArray

    • divi

      public RsNDArray divi(Number n)
      Specified by:
      divi in interface ai.djl.ndarray.NDArray

    • divi

      public RsNDArray divi(ai.djl.ndarray.NDArray other)
      Specified by:
      divi in interface ai.djl.ndarray.NDArray

    • modi

      public RsNDArray modi(Number n)
      Specified by:
      modi in interface ai.djl.ndarray.NDArray

    • modi

      public RsNDArray modi(ai.djl.ndarray.NDArray other)
      Specified by:
      modi in interface ai.djl.ndarray.NDArray

    • powi

      public RsNDArray powi(Number n)
      Specified by:
      powi in interface ai.djl.ndarray.NDArray

    • powi

      public RsNDArray powi(ai.djl.ndarray.NDArray other)
      Specified by:
      powi in interface ai.djl.ndarray.NDArray

    • signi

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

    • negi

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

    • sign

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

    • maximum

      public RsNDArray maximum(Number n)
      Specified by:
      maximum in interface ai.djl.ndarray.NDArray

    • maximum

      public RsNDArray maximum(ai.djl.ndarray.NDArray other)
      Specified by:
      maximum in interface ai.djl.ndarray.NDArray

    • minimum

      public RsNDArray minimum(Number n)
      Specified by:
      minimum in interface ai.djl.ndarray.NDArray

    • minimum

      public RsNDArray minimum(ai.djl.ndarray.NDArray other)
      Specified by:
      minimum in interface ai.djl.ndarray.NDArray

    • all

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

    • any

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

    • none

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

    • countNonzero

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

    • countNonzero

      public ai.djl.ndarray.NDArray countNonzero(int axis)
      Specified by:
      countNonzero in interface ai.djl.ndarray.NDArray

    • neg

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

    • abs

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

    • square

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

    • floor

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

    • ceil

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

    • round

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

    • trunc

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

    • exp

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

    • log10

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

    • log2

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

    • sin

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

    • cos

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

    • tan

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

    • asin

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

    • acos

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

    • atan

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

    • atan2

      public RsNDArray atan2(ai.djl.ndarray.NDArray other)
      Specified by:
      atan2 in interface ai.djl.ndarray.NDArray

    • sinh

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

    • cosh

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

    • tanh

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

    • asinh

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

    • acosh

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

    • atanh

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

    • toDegrees

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

    • toRadians

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

    • max

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

    • max

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

    • min

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

    • min

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

    • sum

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

    • sum

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

    • prod

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

    • mean

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

    • mean

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

    • normalize

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

    • rotate90

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

    • trace

      public RsNDArray 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 RsNDArray 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

    • pad

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

    • reshape

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

    • expandDims

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

    • squeeze

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

    • logicalOr

      public RsNDArray logicalOr(ai.djl.ndarray.NDArray other)
      Specified by:
      logicalOr in interface ai.djl.ndarray.NDArray

    • logicalXor

      public RsNDArray logicalXor(ai.djl.ndarray.NDArray other)
      Specified by:
      logicalXor in interface ai.djl.ndarray.NDArray

    • logicalNot

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

    • argSort

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

    • sort

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

    • sort

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

    • softmax

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

    • logSoftmax

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

    • cumSum

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

    • cumSum

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

    • isNaN

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

    • tile

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

    • tile

      public RsNDArray tile(int axis, long repeat)
      Specified by:
      tile in interface ai.djl.ndarray.NDArray

    • tile

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

    • tile

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

    • repeat

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

    • repeat

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

    • repeat

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

    • repeat

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

    • dot

      public RsNDArray 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 RsNDArray clip(Number min, Number max)
      Specified by:
      clip in interface ai.djl.ndarray.NDArray

    • swapAxes

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

    • transpose

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

    • broadcast

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

    • argMax

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

    • argMax

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

    • argMin

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

    • percentile

      public RsNDArray percentile(Number percentile)
      Specified by:
      percentile in interface ai.djl.ndarray.NDArray

    • percentile

      public RsNDArray percentile(Number percentile, int[] axes)
      Specified by:
      percentile in interface ai.djl.ndarray.NDArray

    • median

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

    • median

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

    • toDense

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

    • toSparse

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

    • nonzero

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

    • erfinv

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

    • erf

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

    • inverse

      public RsNDArray 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, 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

    • getNDArrayInternal

      public RsNDArrayEx 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>