org.tensorflow.lite

Class Tensor

java.lang.Object

org.tensorflow.lite.Tensor

public final class Tensor
extends Object

A typed multi-dimensional array used in Tensorflow Lite.

The native handle of a Tensor is managed by NativeInterpreterWrapper, and does not needed to be closed by the client. However, once the NativeInterpreterWrapper has been closed, the tensor handle will be invalidated.

Method Summary

Modifier and Type	Method and Description
DataType	dataType() Returns the DataType of elements stored in the Tensor.
int	numBytes() Returns the size, in bytes, of the tensor data.
int	numDimensions() Returns the number of dimensions (sometimes referred to as rank) of the Tensor.
int	numElements() Returns the number of elements in a flattened (1-D) view of the tensor.
int[]	shape() Returns the shape of the Tensor, i.e., the sizes of each dimension.

Methods inherited from class java.lang.Object

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

Method Detail

dataType

public DataType dataType()

Returns the DataType of elements stored in the Tensor.

numDimensions

public int numDimensions()

Returns the number of dimensions (sometimes referred to as rank) of the Tensor.

Will be 0 for a scalar, 1 for a vector, 2 for a matrix, 3 for a 3-dimensional tensor etc.

numBytes

public int numBytes()

Returns the size, in bytes, of the tensor data.

numElements

public int numElements()

Returns the number of elements in a flattened (1-D) view of the tensor.

shape

public int[] shape()

Returns the shape of the Tensor, i.e., the sizes of each dimension.

Returns:

an array where the i-th element is the size of the i-th dimension of the tensor.