com.linkedin.feathr.common.tensor.dense

Class DenseIntTensor

java.lang.Object

com.linkedin.feathr.common.tensor.DenseTensor

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

com.linkedin.feathr.common.tensor.dense.DenseIntTensor

All Implemented Interfaces:

TensorData

public class DenseIntTensor
extends ByteBufferDenseTensor

Dense tensor which can be converted from/to a tensorflow tensor object. The underlying data is backed up by IntBuffer.

Field Summary

Fields inherited from class com.linkedin.feathr.common.tensor.dense.ByteBufferDenseTensor

_byteBuffer, _cardinality, _limit, _shape, _start, _types

Constructor Summary

Constructors

Constructor and Description
DenseIntTensor(java.nio.ByteBuffer byteBuffer, long[] shape)
DenseIntTensor(java.nio.ByteBuffer byteBuffer, long[] shape, Representable[] types, int cardinality)

Method Summary

Modifier and Type	Method and Description
java.util.List<?>	asList()
TensorIterator	iterator()

Methods inherited from class com.linkedin.feathr.common.tensor.dense.ByteBufferDenseTensor

cardinality, estimatedCardinality, getByteBuffer, getByteBuffer, getShape, getTypes

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

Constructor Detail

DenseIntTensor

public DenseIntTensor(java.nio.ByteBuffer byteBuffer,
                      long[] shape)

DenseIntTensor

public DenseIntTensor(java.nio.ByteBuffer byteBuffer,
                      long[] shape,
                      Representable[] types,
                      int cardinality)

Method Detail

iterator

public TensorIterator iterator()

Returns:

an iterator over entries in the mapping from dimensions to the value.

asList

public java.util.List<?> asList()

Specified by:

asList in class DenseTensor

Returns:

the list of values in their unraveled order. This in combination with the shape is enough to restore the entire tensor, as the dimensions can be calculated from the shape.