org.nd4j.imports.graphmapper.tf.tensors

Class TFTensorMappers.BaseTensorMapper<T,U extends Buffer>

java.lang.Object

org.nd4j.imports.graphmapper.tf.tensors.TFTensorMappers.BaseTensorMapper<T,U>

All Implemented Interfaces:

TFTensorMapper<T,U>

Direct Known Subclasses:

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

Enclosing class:

TFTensorMappers

public abstract static class TFTensorMappers.BaseTensorMapper<T,U extends Buffer>
extends Object
implements TFTensorMapper<T,U>

Nested Class Summary

Nested classes/interfaces inherited from interface org.nd4j.imports.graphmapper.tf.tensors.TFTensorMapper

TFTensorMapper.ValueSource

Field Summary

Fields

Modifier and Type	Field and Description
protected TensorProto	tfTensor

Constructor Summary

Constructors

Constructor and Description
BaseTensorMapper(TensorProto tensorProto)

Method Summary

Modifier and Type	Method and Description
DataType	dataType()
boolean	isEmpty()
long[]	shape()
INDArray	toNDArray()
TFTensorMapper.ValueSource	valueSource()

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.nd4j.imports.graphmapper.tf.tensors.TFTensorMapper

arrayFor, getBuffer, getValue, getValue, newArray, valueCount

Field Detail

tfTensor

protected TensorProto tfTensor

Constructor Detail

BaseTensorMapper

public BaseTensorMapper(TensorProto tensorProto)

Method Detail

dataType

public DataType dataType()

Specified by:

dataType in interface TFTensorMapper<T,U extends Buffer>

shape

public long[] shape()

Specified by:

shape in interface TFTensorMapper<T,U extends Buffer>

isEmpty

public boolean isEmpty()

Specified by:

isEmpty in interface TFTensorMapper<T,U extends Buffer>

valueSource

public TFTensorMapper.ValueSource valueSource()

Specified by:

valueSource in interface TFTensorMapper<T,U extends Buffer>

toNDArray

public INDArray toNDArray()

Specified by:

toNDArray in interface TFTensorMapper<T,U extends Buffer>