如何在张量流中从张量中提取一个值为int？

Question

这个问题是关于访问张量中的单个元素，比如[1,2,3]。我需要访问内部元素[1]。因为我正在调用另一个不是来自tensorflow的api库。这个api需要一个int值作为参数。我在将[1]放入api时遇到问题，因为[1]显示为张量，而不是int。有没有办法将像[1]这样的单个张量转换为int 1？

Answer 1

您需要启动会话，加载/初始化变量然后运行会话。

# build graph
target_tensor = bigger_tensor[1, 2]  # whichever index you want
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())  # or load from file
    target_value = sess.run(target_tensor)  #may need a feed_dict if your model uses placeholders