com.linkedin.feathr.common.tensor.dense

Class ByteBufferDenseTensor

java.lang.Object

com.linkedin.feathr.common.tensor.DenseTensor

com.linkedin.feathr.common.tensor.dense.ByteBufferDenseTensor

All Implemented Interfaces:

TensorData

Direct Known Subclasses:

DenseBooleanTensor, DenseBytesTensor, DenseDoubleTensor, DenseFloatTensor, DenseIntTensor, DenseLongTensor, DenseStringTensor

public abstract class ByteBufferDenseTensor
extends DenseTensor

Base class for dense int/float/double/long/boolean/string/byte[] tensors backed by a ByteBuffer using TensorFlow-compatible layout.

Field Summary

Fields

Modifier and Type	Field and Description
protected java.nio.ByteBuffer	_byteBuffer
protected int	_cardinality
protected int	_limit
protected long[]	_shape
protected int	_start
protected Representable[]	_types

Constructor Summary

Constructors

Constructor and Description
ByteBufferDenseTensor(java.nio.ByteBuffer byteBuffer, long[] shape, Primitive type, int bytes)

Method Summary

Modifier and Type	Method and Description
int	cardinality()
int	estimatedCardinality()
java.nio.ByteBuffer	getByteBuffer()
java.nio.ByteBuffer	getByteBuffer(boolean makeCopy)
long[]	getShape()
Representable[]	getTypes()

Methods inherited from class com.linkedin.feathr.common.tensor.DenseTensor

asList

Methods inherited from class java.lang.Object

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

Methods inherited from interface com.linkedin.feathr.common.tensor.TensorData

getArity, isEmpty, iterator

Field Detail

_byteBuffer

protected final java.nio.ByteBuffer _byteBuffer

_start

protected final int _start

_limit

protected final int _limit

_shape

protected final long[] _shape

_cardinality

protected final int _cardinality

_types

protected final Representable[] _types

Constructor Detail

ByteBufferDenseTensor

public ByteBufferDenseTensor(java.nio.ByteBuffer byteBuffer,
                             long[] shape,
                             Primitive type,
                             int bytes)

Method Detail

getByteBuffer

public java.nio.ByteBuffer getByteBuffer()

getByteBuffer

public java.nio.ByteBuffer getByteBuffer(boolean makeCopy)

getShape

public long[] getShape()

Returns:

the current shape of the data or the supplied shape of the data

getTypes

public Representable[] getTypes()

Returns:

the types of all dimensions and the value as a single array.

estimatedCardinality

public int estimatedCardinality()

Returns:

conservative estimate of number of entries in the mapping from dimensions to values. Can be cheaper to evaluate than the exact one.

cardinality

public int cardinality()

Returns:

exact number of entries in the mapping from dimensions to values.