Comment imprimer les valeurs des tenseurs TensorFlow ?

Impression des valeurs des tenseurs TensorFlow : un guide complet

Dans TensorFlow, un objet Tensor représente un tableau multidimensionnel de données. Pour accéder aux valeurs réelles stockées dans un Tensor, vous devrez l'évaluer au sein d'une session.

Méthode Session.run()

L'approche la plus simple est pour utiliser la méthode Session.run() pour évaluer le Tensor et récupérer sa valeur :

import tensorflow as tf

sess = tf.Session()
matrix1 = tf.constant([[3., 3.]])
matrix2 = tf.constant([[2.],[2.]])
product = tf.matmul(matrix1, matrix2)
print(sess.run(product))

Cela imprimera la valeur du Tensor sous forme de tableau NumPy.

Tensor Méthode .eval()

Vous pouvez également utiliser la méthode Tensor.eval() pour évaluer un Tensor dans la session par défaut :

with tf.Session():
    print(product.eval())

Session interactive

Pour une approche plus pratique, vous pouvez utiliser tf.InteractiveSession pour ouvrir une session par défaut pour l'ensemble de votre programme :

import tensorflow as tf

tf.InteractiveSession()

matrix1 = tf.constant([[3., 3.]])
matrix2 = tf.constant([[2.],[2.]])
product = tf.matmul(matrix1, matrix2)
print(product.eval())

Notes

• Pour plus d'efficacité, TensorFlow sépare la définition des calculs (construction du graphe de flux de données) de l'exécution (évaluation du graphe et production de valeurs).
• L'opérateur tf.print() peut également être utilisé pour imprime la valeur d'un Tensor, mais cela nécessite une exécution manuelle avec Session.run().
• La fonction tf.get_static_value() peut être utilisée pour obtenir la valeur constante d'un Tensor si elle peut être calculée efficacement.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!