Package 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	Description
protected final ByteBuffer	_byteBuffer
protected final int	_cardinality
protected final int	_limit
protected final long[]	_shape
protected final int	_start
protected final Representable[]	_types

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	Description
int	cardinality()
int	estimatedCardinality()
ByteBuffer	getByteBuffer()
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 Details

_byteBuffer

protected final 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 Details

ByteBufferDenseTensor

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

Method Details

getByteBuffer

public ByteBuffer getByteBuffer()

getByteBuffer

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