Package ai.onnxruntime

Class OnnxTensor

java.lang.Object

ai.onnxruntime.OnnxTensor

All Implemented Interfaces:

OnnxValue, java.lang.AutoCloseable

public class OnnxTensor
extends java.lang.Object
implements OnnxValue

A Java object wrapping an OnnxTensor. Tensors are the main input to the library, and can also be returned as outputs.

Nested Class Summary

Nested classes/interfaces inherited from interface ai.onnxruntime.OnnxValue

OnnxValue.OnnxValueType

Method Summary

Modifier and Type	Method	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.lang.Object data)	Create a Tensor from a Java primitive or String multidimensional array.
static OnnxTensor	createTensor(OrtEnvironment env, java.lang.String[] data, long[] shape)	Create a tensor from a flattened string array.
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.nio.ShortBuffer data, long[] shape)	Create an OnnxTensor backed by a direct ShortBuffer.
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()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Method Detail

getType

public OnnxValue.OnnxValueType getType()

Description copied from interface: OnnxValue

Gets the type of this OnnxValue.

Specified by:

getType in interface OnnxValue

Returns:

The value type.

getValue

public java.lang.Object getValue()
                          throws OrtException

Either returns a boxed primitive if the Tensor is a scalar, or a multidimensional array of primitives if it has multiple dimensions.

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()).

Specified by:

getValue in interface OnnxValue

Returns:

A Java value.

Throws:

OrtException - If the value could not be extracted as the Tensor is invalid, or if the native code encountered an error.

getInfo

public TensorInfo getInfo()

Description copied from interface: OnnxValue

Gets the type info object associated with this OnnxValue.

Specified by:

getInfo in interface OnnxValue

Returns:

The type information.

toString

public java.lang.String toString()

Overrides:

toString in class java.lang.Object

close

public void close()

Closes the tensor, releasing it's underlying memory (if it's not backed by an NIO buffer). If it is backed by a buffer then the memory is released when the buffer is GC'd.

Specified by:

close in interface java.lang.AutoCloseable

Specified by:

close in interface OnnxValue

getByteBuffer

public java.nio.ByteBuffer getByteBuffer()

Returns a copy of the underlying OnnxTensor as a ByteBuffer.

This method returns null if the OnnxTensor contains Strings as they are stored externally to the OnnxTensor.

Returns:

A ByteBuffer copy of the OnnxTensor.

getFloatBuffer

public 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. it's a float or fp16), otherwise it returns null.

Returns:

A FloatBuffer copy of the OnnxTensor.

getDoubleBuffer

public 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.

Returns:

A DoubleBuffer copy of the OnnxTensor.

getShortBuffer

public 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.

Returns:

A ShortBuffer copy of the OnnxTensor.

getIntBuffer

public 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.

Returns:

An IntBuffer copy of the OnnxTensor.

getLongBuffer

public 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.

Returns:

A LongBuffer copy of the OnnxTensor.

createTensor

public static OnnxTensor createTensor(OrtEnvironment env,
                                      java.lang.Object data)
                               throws OrtException

Create a Tensor from a Java primitive or String multidimensional array. The shape is inferred from the array using reflection. The default allocator is used.

Parameters:

env - The current OrtEnvironment.

data - The data to store in a tensor.

Returns:

An OnnxTensor storing the data.

Throws:

OrtException - If the onnx runtime threw an error.

createTensor

public static OnnxTensor createTensor(OrtEnvironment env,
                                      java.lang.String[] data,
                                      long[] shape)
                               throws OrtException

Create a tensor from a flattened string array.

Requires the array to be flattened in row-major order. Uses the default allocator.

Parameters:

env - The current OrtEnvironment.

data - The tensor data

shape - the shape of the tensor

Returns:

An OnnxTensor of the required shape.

Throws:

OrtException - Thrown if there is an onnx error or if the data and shape don't match.

createTensor

public static OnnxTensor createTensor(OrtEnvironment env,
                                      java.nio.FloatBuffer data,
                                      long[] shape)
                               throws OrtException

Create an OnnxTensor backed by a direct FloatBuffer. The buffer should be in nativeOrder.

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.

Parameters:

env - The current OrtEnvironment.

data - The tensor data.

shape - The shape of tensor.

Returns:

An OnnxTensor of the required shape.

Throws:

OrtException - Thrown if there is an onnx error or if the data and shape don't match.

createTensor

public static OnnxTensor createTensor(OrtEnvironment env,
                                      java.nio.DoubleBuffer data,
                                      long[] shape)
                               throws OrtException

Create an OnnxTensor backed by a direct DoubleBuffer. The buffer should be in nativeOrder.

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.

Parameters:

env - The current OrtEnvironment.

data - The tensor data.

shape - The shape of tensor.

Returns:

An OnnxTensor of the required shape.

Throws:

OrtException - Thrown if there is an onnx error or if the data and shape don't match.

createTensor

public static OnnxTensor createTensor(OrtEnvironment env,
                                      java.nio.ByteBuffer data,
                                      long[] shape)
                               throws OrtException

Create an OnnxTensor backed by a direct ByteBuffer. The buffer should be in nativeOrder.

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.

Parameters:

env - The current OrtEnvironment.

data - The tensor data.

shape - The shape of tensor.

Returns:

An OnnxTensor of the required shape.

Throws:

OrtException - Thrown if there is an onnx error or if the data and shape don't match.

createTensor

public static OnnxTensor createTensor(OrtEnvironment env,
                                      java.nio.ByteBuffer data,
                                      long[] shape,
                                      OnnxJavaType type)
                               throws OrtException

Create an OnnxTensor backed by a direct ByteBuffer. The buffer should be in nativeOrder.

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.

Parameters:

env - The current OrtEnvironment.

data - The tensor data.

shape - The shape of tensor.

type - The type to use for the byte buffer elements.

Returns:

An OnnxTensor of the required shape.

Throws:

OrtException - Thrown if there is an onnx error or if the data and shape don't match.

createTensor

public static OnnxTensor createTensor(OrtEnvironment env,
                                      java.nio.ShortBuffer data,
                                      long[] shape)
                               throws OrtException

Create an OnnxTensor backed by a direct ShortBuffer. The buffer should be in nativeOrder.

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.

Parameters:

env - The current OrtEnvironment.

data - The tensor data.

shape - The shape of tensor.

Returns:

An OnnxTensor of the required shape.

Throws:

OrtException - Thrown if there is an onnx error or if the data and shape don't match.

createTensor

public static OnnxTensor createTensor(OrtEnvironment env,
                                      java.nio.IntBuffer data,
                                      long[] shape)
                               throws OrtException

Create an OnnxTensor backed by a direct IntBuffer. The buffer should be in nativeOrder.

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.

Parameters:

env - The current OrtEnvironment.

data - The tensor data.

shape - The shape of tensor.

Returns:

An OnnxTensor of the required shape.

Throws:

OrtException - Thrown if there is an onnx error or if the data and shape don't match.

createTensor

public static OnnxTensor createTensor(OrtEnvironment env,
                                      java.nio.LongBuffer data,
                                      long[] shape)
                               throws OrtException

Create an OnnxTensor backed by a direct LongBuffer. The buffer should be in nativeOrder.

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.

Parameters:

env - The current OrtEnvironment.

data - The tensor data.

shape - The shape of tensor.

Returns:

An OnnxTensor of the required shape.

Throws:

OrtException - Thrown if there is an onnx error or if the data and shape don't match.