tensorflow :: opérations :: Bitcast

#include <array_ops.h>

Bitcasts un tenseur d'un type à un autre sans copier de données.

Résumé

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

Si le type de données d'entrée T est plus grand que le type de type 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 vrai dtype est transtypé comme un dtype complexe (par exemple tf.complex64 ou tf.complex128) comme tf.cast () rend la partie imaginaire 0 tandis que tf.bitcast () donne le module Erreur. Par example,

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)
Exemple 2:
>>> tf.bitcast(tf.constant(0xffffffff, dtype=tf.uint32), tf.uint8)

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

REMARQUE : Bitcast est implémenté en tant que distribution de bas niveau, de sorte que les machines avec des ordres d' extrémité différents donneront des résultats différents.

Arguments:

Retour:

  • 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

production

::tensorflow::Output output

Fonctions publiques

Bitcast

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

nœud

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const