tensorflow::ops::Bitcast
#include <array_ops.h>
Выполняет побитовое приведение (bitcast) тензора из одного типа в другой без копирования данных.
Краткое описание
Для заданного тензора input эта операция возвращает тензор, имеющий те же данные буфера, что и input, с типом данных type.
Если входной тип данных T больше выходного типа данных type, то форма меняется с [...] на [..., sizeof(T)/sizeof(type)].
Если T меньше, чем type, оператор требует, чтобы крайний правый размер был равен sizeof(type)/sizeof(T). В этом случае форма изменяется с [..., sizeof(type)/sizeof(T)] на [...].
tf.bitcast() и tf.cast() работают по-разному, когда вещественный dtype преобразуется в комплексный dtype (например, tf.complex64 или tf.complex128), поскольку tf.cast() делает мнимую часть равной 0, тогда как tf.bitcast() возвращает ошибку модуля. Например,
Пример 1:
a = [1., 2., 3.] equality_bitcast = tf.bitcast(a, tf.complex128) Traceback (most recent call last): ... InvalidArgumentError: Cannot bitcast from 1 to 18 [Op:Bitcast] equality_cast = tf.cast(a, tf.complex128) print(equality_cast) tf.Tensor([1.+0.j 2.+0.j 3.+0.j], shape=(3,), dtype=complex128)
Пример 2:
tf.bitcast(tf.constant(0xffffffff, dtype=tf.uint32), tf.uint8)
Пример 3:
x = [1., 2., 3.] y = [0., 2., 3.] equality= tf.equal(x,y) equality_cast = tf.cast(equality,tf.float32) equality_bitcast = tf.bitcast(equality_cast,tf.uint8) print(equality) tf.Tensor([False True True], shape=(3,), dtype=bool) print(equality_cast) tf.Tensor([0. 1. 1.], shape=(3,), dtype=float32) print(equality_bitcast) tf.Tensor( [[ 0 0 0 0] [ 0 0 128 63] [ 0 0 128 63]], shape=(3, 4), dtype=uint8)
ПРИМЕЧАНИЕ: Bitcast реализован как низкоуровневое приведение типов, поэтому на машинах с разным порядком байтов будут получаться разные результаты. На машинах с прямым порядком байтов (BE) при несовпадении размеров типов выполняется копирование из входного буфера в выходной для получения тех же результатов приведения, что и на машинах с обратным порядком байтов (LE).
Аргументы:
- scope: Объект Scope
Возвращает:
-
Output: Выходной тензор.
| Конструкторы и деструкторы | |
|---|---|
Bitcast(const ::tensorflow::Scope & scope, ::tensorflow::Input input, DataType type) |
| Открытые атрибуты | |
|---|---|
operation | |
output | |
| Открытые функции | |
|---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const | |
Открытые атрибуты
operation
Operation operation
output
::tensorflow::Output output
Открытые функции
Bitcast
Bitcast( const ::tensorflow::Scope & scope, ::tensorflow::Input input, DataType type )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/bitcast