降低Tensorflow的一张一张量的等级?

时间:2018-11-14 04:33:52

标签: python tensorflow

我有一个Tensorflow张量[[1]],如何将这个张量更改为只是一个数字的[1]?

非常感谢您!

1 个答案:

答案 0 :(得分:0)

在这里,我创建了一个虚拟2D张量并将其转换为您要查找的标量。请注意,您可以在会话tf.Session().run(tensor)内而不是后面的InteractiveSession内运行此程序。

这里要注意的一件事是,Session.runeval的任何返回张量都是一个numpy数组。如果返回的数组是多维数组,则可以调用np.ravel()将其设为1D,并使用第一个元素将其设为标量。

>>> sess = tf.InteractiveSession()
>>> t=[[10]]
>>> tensor=tf.convert_to_tensor(t)
>>> tensor.eval().ravel()[0]
10