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.NDArrayAdapter
OrtNDArray
is the ONNX Runtime implementation ofNDArray
.
-
-
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 theOnnxTensor
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.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
-
-
-
-
Method Detail
-
getTensor
public ai.onnxruntime.OnnxTensor getTensor()
Returns theOnnxTensor
representation of this OrtNDArray.- Returns:
- the
OnnxTensor
representation of this OrtNDArray
-
getDataType
public ai.djl.ndarray.types.DataType getDataType()
- Specified by:
getDataType
in interfaceai.djl.ndarray.NDArray
- Overrides:
getDataType
in classai.djl.ndarray.NDArrayAdapter
-
getShape
public ai.djl.ndarray.types.Shape getShape()
- Specified by:
getShape
in interfaceai.djl.ndarray.NDArray
- Overrides:
getShape
in 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:
toStringArray
in interfaceai.djl.ndarray.NDArray
- Overrides:
toStringArray
in classai.djl.ndarray.NDArrayAdapter
-
toByteBuffer
public java.nio.ByteBuffer toByteBuffer()
-
close
public void close()
- Specified by:
close
in interfacejava.lang.AutoCloseable
- Specified by:
close
in interfaceai.djl.ndarray.NDArray
- Specified by:
close
in interfaceai.djl.ndarray.NDResource
- Overrides:
close
in classai.djl.ndarray.NDArrayAdapter
-
-