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.] Egalité_bitcast = tf.bitcast(a, tf.complex128) Traceback (dernier appel le plus récent) : ... InvalidArgumentError : Impossible de diffuser le bit 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.

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