org.nd4j.imports.graphmapper.tf.tensors

Interface TFTensorMapper<J,B extends Buffer>

Type Parameters:

J - Java array type

B - Java buffer type

All Known Implementing Classes:

TFTensorMappers.BaseTensorMapper, TFTensorMappers.BFloat16TensorMapper, TFTensorMappers.BoolTensorMapper, TFTensorMappers.Float16TensorMapper, TFTensorMappers.Float32TensorMapper, TFTensorMappers.Float64TensorMapper, TFTensorMappers.Int16TensorMapper, TFTensorMappers.Int32TensorMapper, TFTensorMappers.Int64TensorMapper, TFTensorMappers.Int8TensorMapper, TFTensorMappers.StringTensorMapper, TFTensorMappers.UInt16TensorMapper, TFTensorMappers.UInt32TensorMapper, TFTensorMappers.UInt64TensorMapper, TFTensorMappers.UInt8TensorMapper

public interface TFTensorMapper<J,B extends Buffer>

Nested Class Summary

Nested Classes

Modifier and Type	Interface and Description
static class	TFTensorMapper.ValueSource

Method Summary

Modifier and Type	Method and Description
INDArray	arrayFor(long[] shape, J jArr)
DataType	dataType()
B	getBuffer(ByteBuffer bb)
void	getValue(J jArr, B buffer, int i)
void	getValue(J jArr, int i)
boolean	isEmpty()
J	newArray(int length)
long[]	shape()
INDArray	toNDArray()
int	valueCount()
TFTensorMapper.ValueSource	valueSource()

Method Detail

dataType

DataType dataType()

shape

long[] shape()

isEmpty

boolean isEmpty()

valueSource

TFTensorMapper.ValueSource valueSource()

valueCount

int valueCount()

newArray

J newArray(int length)

getBuffer

B getBuffer(ByteBuffer bb)

toNDArray

INDArray toNDArray()

getValue

void getValue(J jArr,
              int i)

getValue

void getValue(J jArr,
              B buffer,
              int i)

arrayFor

INDArray arrayFor(long[] shape,
                  J jArr)