张量流跟踪不起作用

Question

我是 TensorFlow 的新手。我有一个张量 x 如下：

>>> x.eval()
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]], dtype=int32)

我想计算这个张量的迹，所以我使用tf.trace，但是它抛出了错误：

>>> tf.trace(x)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: 'module' object has no attribute 'trace'

我该如何解决？非常感谢

Answer 1

为我工作

import tensorflow as tf
x = tf.constant([[1, 2, 3],[4, 5, 6],[7, 8, 9]], dtype=tf.int32)
sess = tf.InteractiveSession()
x.eval()
tf.trace(x).eval()

打印出来

[[1 2 3]
 [4 5 6]
 [7 8 9]]
15