@TensorType(dataType=DT_HALF, byteSize=2, mapperClass=TFloat16Mapper.class) public interface TFloat16 extends org.tensorflow.ndarray.FloatNdArray, TFloating
Since there is no floating-point type that fits in 16 bits in Java, a conversion (with
potentially a precision loss) is required for each 32 bits value written or read on a tensor of
this type from the JVM. Therefore, if a lot of I/O operations are to be expected on a tensor,
performances will be improved by working with TFloat32
or TFloat64
data types
whenever possible.
Also, TFloat16
tensors normally perform better if they are located in GPU memory since
most CPUs do not support this format natively. For CPU computation on 16-bit floats, the TBfloat16
tensor type might be a better option.
Modifier and Type | Method and Description |
---|---|
static TFloat16 |
scalarOf(float value)
Allocates a new tensor for storing a single float value.
|
static TFloat16 |
tensorOf(org.tensorflow.ndarray.NdArray<Float> src)
Allocates a new tensor which is a copy of a given array of floats.
|
static TFloat16 |
tensorOf(org.tensorflow.ndarray.Shape shape)
Allocates a new tensor of the given shape.
|
static TFloat16 |
tensorOf(org.tensorflow.ndarray.Shape shape,
Consumer<TFloat16> dataInit)
Allocates a new tensor of the given shape and initialize its data.
|
static TFloat16 |
tensorOf(org.tensorflow.ndarray.Shape shape,
org.tensorflow.ndarray.buffer.FloatDataBuffer data)
Allocates a new tensor of the given shape, initialized with the provided data.
|
static TFloat16 |
vectorOf(float... values)
Allocates a new tensor for storing a vector of floats.
|
static TFloat16 scalarOf(float value)
value
- float to store in the new tensorstatic TFloat16 vectorOf(float... values)
values
- floats to store in the new tensorstatic TFloat16 tensorOf(org.tensorflow.ndarray.NdArray<Float> src)
The tensor will have the same shape as the source array and its data will be copied.
src
- the source array giving the shape and data to the new tensorstatic TFloat16 tensorOf(org.tensorflow.ndarray.Shape shape)
shape
- shape of the tensor to allocatestatic TFloat16 tensorOf(org.tensorflow.ndarray.Shape shape, org.tensorflow.ndarray.buffer.FloatDataBuffer data)
shape
- shape of the tensor to allocatedata
- buffer of floats to initialize the tensor withstatic TFloat16 tensorOf(org.tensorflow.ndarray.Shape shape, Consumer<TFloat16> dataInit)
shape
- shape of the tensor to allocatedataInit
- tensor data initializerTensorFlowException
- if the tensor cannot be allocated or initializedCopyright © 2015–2022. All rights reserved.