Class OrtNDArray

  • All Implemented Interfaces:
    ai.djl.ndarray.BytesSupplier, ai.djl.ndarray.NDArray, ai.djl.ndarray.NDResource, java.lang.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

        alternativeManager, dataType, isClosed, manager, name, shape, uid
    • Method Summary

      All Methods Instance Methods Concrete Methods 
      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)
      java.nio.ByteBuffer toByteBuffer()
      java.lang.String[] toStringArray​(java.nio.charset.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, atanh, attach, batchDot, batchMatMul, booleanMask, broadcast, cbrt, ceil, clip, complex, contentEquals, contentEquals, cos, cosh, cumProd, cumProd, cumSum, cumSum, div, div, divi, divi, dot, eq, eq, equals, erfinv, exp, expandDims, fft, flatten, flatten, flip, floor, gammaln, gather, gatherNd, get, getDevice, getGradient, getManager, getName, getNDArrayInternal, getSparseFormat, getUid, gt, gt, gte, gte, hasGradient, hashCode, inverse, 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, percentile, percentile, pow, pow, powi, powi, prod, prod, put, real, repeat, repeat, repeat, repeat, reshape, 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, get, get, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, getResourceNDArrays, getScalar, getUint8, isEmpty, isScalar, isSparse, like, max, mean, min, none, norm, norm, norm, normalize, normalize, oneHot, oneHot, onesLike, prod, reshape, scaleGradient, set, set, set, set, set, shapeEquals, size, size, split, split, squeeze, squeeze, stack, stack, stft, sum, swapAxes, take, toArray, toBooleanArray, toByteArray, 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 Detail

      • 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 java.lang.String[] toStringArray​(java.nio.charset.Charset charset)
        Specified by:
        toStringArray in interface ai.djl.ndarray.NDArray
        Overrides:
        toStringArray in class ai.djl.ndarray.NDArrayAdapter
      • toByteBuffer

        public java.nio.ByteBuffer toByteBuffer()
      • 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.ndarray.NDArrayAdapter