org.tribuo.interop.tensorflow

Class TensorFlowUtil.TensorTuple

java.lang.Object

org.tribuo.interop.tensorflow.TensorFlowUtil.TensorTuple

All Implemented Interfaces:

Serializable

Enclosing class:

TensorFlowUtil

public static final class TensorFlowUtil.TensorTuple
extends Object
implements Serializable

A serializable tuple containing the tensor class name, the shape and the data.

It's almost a record.

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
String	className
byte[]	data
long[]	shape

Constructor Summary

Constructors

Constructor and Description
TensorTuple(String className, long[] shape, byte[] data) Makes a TensorTuple.

Method Summary

Modifier and Type	Method and Description
static TensorFlowUtil.TensorTuple	of(org.tensorflow.types.family.TType tensor) Makes a TensorTuple out of this tensor.
org.tensorflow.Tensor	rebuildTensor() Recreates the Tensor from the serialized form.

Methods inherited from class java.lang.Object

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

Field Detail

className

public final String className

shape

public final long[] shape

data

public final byte[] data

Constructor Detail

TensorTuple

public TensorTuple(String className,
                   long[] shape,
                   byte[] data)

Makes a TensorTuple.

Parameters:

className - The tensor class name.

shape - The dimensions of the tensor.

data - The data in the tensor.

Method Detail

rebuildTensor

public org.tensorflow.Tensor rebuildTensor()

Recreates the Tensor from the serialized form.

Returns:

The Tensor.

of

public static TensorFlowUtil.TensorTuple of(org.tensorflow.types.family.TType tensor)

Makes a TensorTuple out of this tensor.

Parameters:

tensor - The tensor to serialize.

Returns:

A serializable form of the Tensor.