OnnxValue.OnnxValueType
Modifier and Type | Method and Description |
---|---|
void |
close()
Closes the tensor, releasing it's underlying memory (if it's not backed by an NIO buffer).
|
static OnnxTensor |
createTensor(OrtEnvironment env,
java.nio.ByteBuffer data,
long[] shape)
Create an OnnxTensor backed by a direct ByteBuffer.
|
static OnnxTensor |
createTensor(OrtEnvironment env,
java.nio.ByteBuffer data,
long[] shape,
OnnxJavaType type)
Create an OnnxTensor backed by a direct ByteBuffer.
|
static OnnxTensor |
createTensor(OrtEnvironment env,
java.nio.DoubleBuffer data,
long[] shape)
Create an OnnxTensor backed by a direct DoubleBuffer.
|
static OnnxTensor |
createTensor(OrtEnvironment env,
java.nio.FloatBuffer data,
long[] shape)
Create an OnnxTensor backed by a direct FloatBuffer.
|
static OnnxTensor |
createTensor(OrtEnvironment env,
java.nio.IntBuffer data,
long[] shape)
Create an OnnxTensor backed by a direct IntBuffer.
|
static OnnxTensor |
createTensor(OrtEnvironment env,
java.nio.LongBuffer data,
long[] shape)
Create an OnnxTensor backed by a direct LongBuffer.
|
static OnnxTensor |
createTensor(OrtEnvironment env,
java.lang.Object data)
Create a Tensor from a Java primitive or String multidimensional array.
|
static OnnxTensor |
createTensor(OrtEnvironment env,
java.nio.ShortBuffer data,
long[] shape)
Create an OnnxTensor backed by a direct ShortBuffer.
|
static OnnxTensor |
createTensor(OrtEnvironment env,
java.lang.String[] data,
long[] shape)
Create a tensor from a flattened string array.
|
java.nio.ByteBuffer |
getByteBuffer()
Returns a copy of the underlying OnnxTensor as a ByteBuffer.
|
java.nio.DoubleBuffer |
getDoubleBuffer()
Returns a copy of the underlying OnnxTensor as a DoubleBuffer if the underlying type is a
double, otherwise it returns null.
|
java.nio.FloatBuffer |
getFloatBuffer()
Returns a copy of the underlying OnnxTensor as a FloatBuffer if it can be losslessly converted
into a float (i.e.
|
TensorInfo |
getInfo()
Gets the type info object associated with this OnnxValue.
|
java.nio.IntBuffer |
getIntBuffer()
Returns a copy of the underlying OnnxTensor as an IntBuffer if the underlying type is int32 or
uint32, otherwise it returns null.
|
java.nio.LongBuffer |
getLongBuffer()
Returns a copy of the underlying OnnxTensor as a LongBuffer if the underlying type is int64 or
uint64, otherwise it returns null.
|
java.nio.ShortBuffer |
getShortBuffer()
Returns a copy of the underlying OnnxTensor as a ShortBuffer if the underlying type is int16 or
uint16, otherwise it returns null.
|
OnnxValue.OnnxValueType |
getType()
Gets the type of this OnnxValue.
|
java.lang.Object |
getValue()
Either returns a boxed primitive if the Tensor is a scalar, or a multidimensional array of
primitives if it has multiple dimensions.
|
java.lang.String |
toString() |
public OnnxValue.OnnxValueType getType()
OnnxValue
public java.lang.Object getValue() throws OrtException
Java multidimensional arrays are quite slow for more than 2 dimensions, in that case it is
recommended you use the java.nio.Buffer extractors below (e.g. getFloatBuffer()
).
getValue
in interface OnnxValue
OrtException
- If the value could not be extracted as the Tensor is invalid, or if the
native code encountered an error.public TensorInfo getInfo()
OnnxValue
public java.lang.String toString()
toString
in class java.lang.Object
public void close()
public java.nio.ByteBuffer getByteBuffer()
This method returns null if the OnnxTensor contains Strings as they are stored externally to the OnnxTensor.
public java.nio.FloatBuffer getFloatBuffer()
public java.nio.DoubleBuffer getDoubleBuffer()
public java.nio.ShortBuffer getShortBuffer()
public java.nio.IntBuffer getIntBuffer()
public java.nio.LongBuffer getLongBuffer()
public static OnnxTensor createTensor(OrtEnvironment env, java.lang.Object data) throws OrtException
env
- The current OrtEnvironment.data
- The data to store in a tensor.OrtException
- If the onnx runtime threw an error.public static OnnxTensor createTensor(OrtEnvironment env, java.lang.String[] data, long[] shape) throws OrtException
Requires the array to be flattened in row-major order. Uses the default allocator.
env
- The current OrtEnvironment.data
- The tensor datashape
- the shape of the tensorOrtException
- Thrown if there is an onnx error or if the data and shape don't match.public static OnnxTensor createTensor(OrtEnvironment env, java.nio.FloatBuffer data, long[] shape) throws OrtException
If the supplied buffer is not a direct buffer, a direct copy is created tied to the lifetime of the tensor. Uses the default allocator.
env
- The current OrtEnvironment.data
- The tensor data.shape
- The shape of tensor.OrtException
- Thrown if there is an onnx error or if the data and shape don't match.public static OnnxTensor createTensor(OrtEnvironment env, java.nio.DoubleBuffer data, long[] shape) throws OrtException
If the supplied buffer is not a direct buffer, a direct copy is created tied to the lifetime of the tensor. Uses the default allocator.
env
- The current OrtEnvironment.data
- The tensor data.shape
- The shape of tensor.OrtException
- Thrown if there is an onnx error or if the data and shape don't match.public static OnnxTensor createTensor(OrtEnvironment env, java.nio.ByteBuffer data, long[] shape) throws OrtException
If the supplied buffer is not a direct buffer, a direct copy is created tied to the lifetime
of the tensor. Uses the default allocator. Tells the runtime it's OnnxJavaType.INT8
.
env
- The current OrtEnvironment.data
- The tensor data.shape
- The shape of tensor.OrtException
- Thrown if there is an onnx error or if the data and shape don't match.public static OnnxTensor createTensor(OrtEnvironment env, java.nio.ByteBuffer data, long[] shape, OnnxJavaType type) throws OrtException
If the supplied buffer is not a direct buffer, a direct copy is created tied to the lifetime of the tensor. Uses the default allocator. Tells the runtime it's the specified type.
env
- The current OrtEnvironment.data
- The tensor data.shape
- The shape of tensor.type
- The type to use for the byte buffer elements.OrtException
- Thrown if there is an onnx error or if the data and shape don't match.public static OnnxTensor createTensor(OrtEnvironment env, java.nio.ShortBuffer data, long[] shape) throws OrtException
If the supplied buffer is not a direct buffer, a direct copy is created tied to the lifetime of the tensor. Uses the default allocator.
env
- The current OrtEnvironment.data
- The tensor data.shape
- The shape of tensor.OrtException
- Thrown if there is an onnx error or if the data and shape don't match.public static OnnxTensor createTensor(OrtEnvironment env, java.nio.IntBuffer data, long[] shape) throws OrtException
If the supplied buffer is not a direct buffer, a direct copy is created tied to the lifetime of the tensor. Uses the default allocator.
env
- The current OrtEnvironment.data
- The tensor data.shape
- The shape of tensor.OrtException
- Thrown if there is an onnx error or if the data and shape don't match.public static OnnxTensor createTensor(OrtEnvironment env, java.nio.LongBuffer data, long[] shape) throws OrtException
If the supplied buffer is not a direct buffer, a direct copy is created tied to the lifetime of the tensor. Uses the default allocator.
env
- The current OrtEnvironment.data
- The tensor data.shape
- The shape of tensor.OrtException
- Thrown if there is an onnx error or if the data and shape don't match.