Package ai.djl.onnxruntime.engine

Class OrtNDArray

java.lang.Object
ai.djl.ndarray.NDArrayAdapter
ai.djl.onnxruntime.engine.OrtNDArray
All Implemented Interfaces:
ai.djl.ndarray.BytesSupplier, ai.djl.ndarray.NDArray, ai.djl.ndarray.NDResource, AutoCloseable
public class OrtNDArray extends ai.djl.ndarray.NDArrayAdapter
OrtNDArray is the ONNX Runtime implementation of NDArray.

  • Field Summary

    Fields inherited from class ai.djl.ndarray.NDArrayAdapter

    alternativeArray, alternativeManager, dataType, isClosed, manager, name, shape, uid

  • Method Summary

    Modifier and Type
    Method
    Description
    void
    close()
    void
    detach()
    ai.djl.ndarray.types.DataType
    getDataType()
    ai.djl.ndarray.types.Shape
    getShape()
    ai.onnxruntime.OnnxTensor
    getTensor()
    Returns the OnnxTensor representation of this OrtNDArray.
    void
    intern(ai.djl.ndarray.NDArray replaced)
    ByteBuffer
    toByteBuffer(boolean tryDirect)
    String[]
    toStringArray(Charset charset)

    Methods inherited from class ai.djl.ndarray.NDArrayAdapter

    abs, acos, acosh, add, add, addi, addi, argMax, argMax, argMin, argMin, argSort, asin, asinh, atan, atan2, atanh, attach, batchDot, batchMatMul, booleanMask, broadcast, cbrt, ceil, clip, complex, conj, contentEquals, contentEquals, cos, cosh, cumProd, cumProd, cumSum, cumSum, diagonal, diagonal, diagonal, diff, div, div, divi, divi, dot, eq, eq, equals, erf, erfinv, exp, expandDims, fft, fft2, flatten, flatten, flip, floor, gammaln, gather, gatherNd, get, getDevice, getGradient, getManager, getName, getNDArrayInternal, getSparseFormat, getUid, gt, gt, gte, gte, hasGradient, hashCode, ifft, ifft2, inverse, irfft, isInfinite, isNaN, isReleased, log, log10, log2, logicalAnd, logicalNot, logicalOr, logicalXor, logSoftmax, lt, lt, lte, lte, matMul, max, max, maximum, maximum, mean, mean, median, median, min, min, minimum, minimum, mod, mod, modi, modi, mul, mul, muli, muli, neg, negi, neq, neq, nonzero, norm, norm, normalize, oneHot, pad, percentile, percentile, pow, pow, powi, powi, prod, prod, put, real, repeat, repeat, repeat, repeat, reshape, rfft, rotate90, round, scatter, sequenceMask, sequenceMask, set, set, set, set, set, setName, setRequiresGradient, setScalar, sign, signi, sin, sinh, softmax, sort, sort, split, split, sqrt, square, squeeze, stft, stopGradient, sub, sub, subi, subi, sum, sum, take, tan, tanh, tempAttach, tile, tile, tile, tile, toDegrees, toDense, toDevice, topK, toRadians, toSparse, toString, toType, trace, transpose, transpose, trunc, unique, xlogy

    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

    all, allClose, allClose, any, argSort, argSort, booleanMask, broadcast, concat, concat, copyTo, countNonzero, countNonzero, duplicate, encode, fft, fft2, get, get, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, getResourceNDArrays, getScalar, getUint8, ifft, ifft2, irfft, isEmpty, isScalar, isSparse, like, max, mean, min, none, norm, norm, norm, normalize, normalize, oneHot, oneHot, onesLike, prod, reshape, rfft, scaleGradient, set, set, set, set, set, shapeEquals, size, size, split, split, squeeze, squeeze, stack, stack, stft, sum, swapAxes, take, toArray, toBooleanArray, toByteArray, toByteBuffer, toDebugString, toDebugString, toDebugString, toDoubleArray, toFloatArray, toIntArray, toLongArray, topK, toShortArray, toStringArray, toUint8Array, toUnsignedIntArray, toUnsignedShortArray, trace, trace, unique, unique, zerosLike

    Methods inherited from interface ai.djl.ndarray.NDResource

    returnResource

  • Method Details

    • getTensor

      public ai.onnxruntime.OnnxTensor getTensor()
      Returns the OnnxTensor representation of this OrtNDArray.
      Returns:
      the OnnxTensor representation of this OrtNDArray

    • getDataType

      public ai.djl.ndarray.types.DataType getDataType()
      Specified by:
      getDataType in interface ai.djl.ndarray.NDArray
      Overrides:
      getDataType in class ai.djl.ndarray.NDArrayAdapter

    • getShape

      public ai.djl.ndarray.types.Shape getShape()
      Specified by:
      getShape in interface ai.djl.ndarray.NDArray
      Overrides:
      getShape in class ai.djl.ndarray.NDArrayAdapter

    • intern

      public void intern(ai.djl.ndarray.NDArray replaced)

    • detach

      public void detach()

    • toStringArray

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

    • toByteBuffer

      public ByteBuffer toByteBuffer(boolean tryDirect)

    • 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.ndarray.NDArrayAdapter