flux tensoriel : : opérations : : Bitcast

#include <array_ops.h>

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.] Egalité_bitcast = tf.bitcast(a, tf.complex128) Traceback (dernier appel le plus récent) : ... InvalidArgumentError : Impossible de diffuser des bits de 1 à 18 [Op : Bitcast ] Egalité_cast = tf.cast(a, tf.complex128) print(equality_cast) tf.Tensor([1.+0.j 2.+0.j 3.+0.j], shape=(3,), dtype=complex128)

Exemple 2 :

tf.bitcast(tf.constant(0xffffffff, dtype=tf.uint32), tf.uint8)

Exemple 3 :

x = [1., 2., 3.] y = [0., 2., 3.] égalité= tf.equal(x,y) égalité_cast = tf.cast(égalité,tf.float32) égalité_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.

Constructeurs et Destructeurs

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

Attributs publics

operation
output

Fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Attributs publics

opération 

Operation operation

sortir 

::tensorflow::Output output

Fonctions publiques

Bitcast 

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

nœud 

::tensorflow::Node * node() const

opérateur :: tensorflow :: Entrée 

 operator::tensorflow::Input() const

opérateur :: tensorflow :: Sortie 

 operator::tensorflow::Output() const