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,java.lang.AutoCloseable
public class OrtNDArray extends ai.djl.ndarray.NDArrayAdapterOrtNDArrayis the ONNX Runtime implementation ofNDArray.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description voidclose()voiddetach()ai.djl.ndarray.types.DataTypegetDataType()ai.djl.ndarray.types.ShapegetShape()ai.onnxruntime.OnnxTensorgetTensor()Returns theOnnxTensorrepresentation of this OrtNDArray.voidintern(ai.djl.ndarray.NDArray replaced)java.nio.ByteBuffertoByteBuffer()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, booleanMask, broadcast, cbrt, ceil, clip, contentEquals, contentEquals, copyTo, cos, cosh, cumProd, cumProd, cumSum, cumSum, div, div, divi, divi, dot, eq, eq, erfinv, exp, expandDims, flatten, flip, floor, gammaln, gather, get, getDevice, getGradient, getManager, getName, getNDArrayInternal, getSparseFormat, getUid, gt, gt, gte, gte, hasGradient, inverse, isInfinite, isNaN, 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, repeat, repeat, repeat, repeat, reshape, rotate90, round, sequenceMask, sequenceMask, set, set, set, set, set, setName, setRequiresGradient, setScalar, sign, signi, sin, sinh, softmax, sort, sort, split, split, sqrt, square, squeeze, stopGradient, sub, sub, subi, subi, sum, sum, take, tan, tanh, tempAttach, tile, tile, tile, tile, toDegrees, toDense, toDevice, toRadians, toSparse, toString, toType, trace, transpose, transpose, trunc
-
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait
-
Methods inherited from interface ai.djl.ndarray.NDArray
all, allClose, allClose, any, argSort, argSort, booleanMask, broadcast, concat, concat, countNonzero, countNonzero, duplicate, encode, 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, sum, swapAxes, take, toArray, toBooleanArray, toByteArray, toDebugString, toDebugString, toDebugString, toDoubleArray, toFloatArray, toIntArray, toLongArray, toStringArray, toUint8Array, trace, trace, zerosLike
-
-
-
-
Method Detail
-
getTensor
public ai.onnxruntime.OnnxTensor getTensor()
Returns theOnnxTensorrepresentation of this OrtNDArray.- Returns:
- the
OnnxTensorrepresentation of this OrtNDArray
-
getDataType
public ai.djl.ndarray.types.DataType getDataType()
- Specified by:
getDataTypein interfaceai.djl.ndarray.NDArray- Overrides:
getDataTypein classai.djl.ndarray.NDArrayAdapter
-
getShape
public ai.djl.ndarray.types.Shape getShape()
- Specified by:
getShapein interfaceai.djl.ndarray.NDArray- Overrides:
getShapein classai.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:
toStringArrayin interfaceai.djl.ndarray.NDArray- Overrides:
toStringArrayin classai.djl.ndarray.NDArrayAdapter
-
toByteBuffer
public java.nio.ByteBuffer toByteBuffer()
-
close
public void close()
- Specified by:
closein interfacejava.lang.AutoCloseable- Specified by:
closein interfaceai.djl.ndarray.NDArray- Specified by:
closein interfaceai.djl.ndarray.NDResource- Overrides:
closein classai.djl.ndarray.NDArrayAdapter
-
-