Spec-Zone.ru › TensorFlow C++

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
Operation
output
::tensorflow::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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API