flux tensoriel : : opérations : : Bitcast

Bitcaste un tenseur d'un type à un autre sans copier les données.

Résumé

Étant donné une input de tenseur, cette opération renvoie un tenseur qui a les mêmes données de tampon que input avec le type de type de données.

Si le type de données d'entrée T est plus grand que le type de données de sortie, la forme passe de [...] à [..., sizeof( T )/sizeof( type )].

Si T est plus petit que type , l'opérateur exige que la dimension la plus à droite soit égale à sizeof( type )/sizeof( T ). La forme passe alors de [..., sizeof( type )/sizeof( T )] à [...].

tf.bitcast() et tf.cast() fonctionnent différemment lorsque le type réel est converti en type complexe (par exemple tf.complex64 ou tf.complex128) car tf.cast() crée une partie imaginaire 0 tandis que tf.bitcast() donne le module erreur. Par exemple,

Exemple 1 :

>>> a = [1., 2., 3.]
>>> equality_bitcast = tf.bitcast(a,tf.complex128)
tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot bitcast from float to complex128: shape [3] [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)

>>> tf.bitcast(tf.constant(0xffffffff, dtype=tf.uint32), tf.uint8)
<tf.Tensor: ... shape=(4,), dtype=uint8, numpy=array([255, 255, 255, 255], dtype=uint8)>

>>> 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)

REMARQUE : Bitcast est implémenté comme une distribution de bas niveau, donc les machines avec des ordres endian différents donneront des résultats différents.

Arguments :

• scope : un objet Scope

Retours :

• Output : Le tenseur de sortie.

Attributs publics

Operation operation

::tensorflow::Output output

 Bitcast(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  DataType type
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const