我有一个Tensorflow张量[[1]],如何将这个张量更改为只是一个数字的[1]?
非常感谢您!
答案 0 :(得分:0)
在这里,我创建了一个虚拟2D张量并将其转换为您要查找的标量。请注意,您可以在会话tf.Session().run(tensor)
内而不是后面的InteractiveSession
内运行此程序。
这里要注意的一件事是,Session.run
或eval
的任何返回张量都是一个numpy数组。如果返回的数组是多维数组,则可以调用np.ravel()将其设为1D,并使用第一个元素将其设为标量。
>>> sess = tf.InteractiveSession()
>>> t=[[10]]
>>> tensor=tf.convert_to_tensor(t)
>>> tensor.eval().ravel()[0]
10