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 Description protected TensorProto
tfTensor
-
Constructor Summary
Constructors Constructor Description BaseTensorMapper(TensorProto tensorProto)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method 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 interfaceTFTensorMapper<T,U extends Buffer>
-
shape
public long[] shape()
- Specified by:
shape
in interfaceTFTensorMapper<T,U extends Buffer>
-
isEmpty
public boolean isEmpty()
- Specified by:
isEmpty
in interfaceTFTensorMapper<T,U extends Buffer>
-
valueSource
public TFTensorMapper.ValueSource valueSource()
- Specified by:
valueSource
in interfaceTFTensorMapper<T,U extends Buffer>
-
-