org.tensorflow

Class RawTensor

java.lang.Object

org.tensorflow.RawTensor

All Implemented Interfaces:

AutoCloseable, org.tensorflow.ndarray.Shaped, Tensor

public final class RawTensor
extends Object
implements Tensor

A tensor which memory has not been mapped to a data space directly accessible from the JVM.

A raw tensor is a minimalist representation of a tensor allocated in native memory by the TensorFlow runtime library and it controls its lifetime within the current process. The data is represented by a flat buffer of bytes, until it is mapped in a n-dimensional typed space by a typed tensor.

Instances of a RawTensor are not thread-safe and their resource must be released by calling close() explicitly or implicitly via try-with-resources.

Method Summary

Modifier and Type	Method and Description
RawTensor	asRawTensor() Returns a raw (untyped) representation of this tensor
void	close() Release resources associated with the Tensor.
org.tensorflow.ndarray.buffer.ByteDataBuffer	data() Returns the raw data of this tensor as a buffer of bytes.
DataType	dataType() Returns the DataType of elements stored in the tensor.
long	numBytes() Returns the size, in bytes, of the tensor data.
org.tensorflow.ndarray.Shape	shape() Returns the shape of the tensor.
String	toString() Returns a string describing the type and shape of the tensor.

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.tensorflow.Tensor

of, of, of, of, of

Methods inherited from interface org.tensorflow.ndarray.Shaped

rank, size

Method Detail

dataType

public DataType dataType()

Description copied from interface: Tensor

Returns the DataType of elements stored in the tensor.

Specified by:

dataType in interface Tensor

numBytes

public long numBytes()

Description copied from interface: Tensor

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

Specified by:

numBytes in interface Tensor

shape

public org.tensorflow.ndarray.Shape shape()

Description copied from interface: Tensor

Returns the shape of the tensor.

Specified by:

shape in interface org.tensorflow.ndarray.Shaped

Specified by:

shape in interface Tensor

asRawTensor

public RawTensor asRawTensor()

Description copied from interface: Tensor

Returns a raw (untyped) representation of this tensor

Specified by:

asRawTensor in interface Tensor

close

public void close()

Description copied from interface: Tensor

Release resources associated with the Tensor.

WARNING:This must be invoked for all tensors that were not been produced by an eager operation or memory will be leaked.

The Tensor object is no longer usable after close returns.

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Tensor

data

public org.tensorflow.ndarray.buffer.ByteDataBuffer data()

Returns the raw data of this tensor as a buffer of bytes.

Returns:

the tensor bytes

Throws:

IllegalStateException - if the tensor has been closed

toString

public String toString()

Returns a string describing the type and shape of the tensor.

Overrides:

toString in class Object