python - 如何迭代tensorflow，python中每個嵌套張量的第一個元素？

Question

我正在使用一個如下所示的張量：

X = tf.constant([['a', 'y', 'b'],
              ['b', 'y', 'a'],
              ['a', 'y', 'c'],
              ['c', 'y', 'a'],
              ['a', 'y', 'd'],
              ['c', 'y', 'd'],
              ['b', 'y', 'c'],
              ['f', 'y', 'e']])

我希望以一種能夠檢索每個嵌套張量的第一個元素的方式進行迭代，即“a”、“b”、“a”、“c”……並在其中執行一些操作迭代。

我曾嘗試使用 tf.slice() 操作，但我是 tensorflow 的新手，無法弄清楚如何去做。 任何幫助將不勝感激。 謝謝！

Answer 1

您可能還沒有評估張量。 使用tensor.eval()或session.run(tensor)來評估結果：

import tensorflow as tf

X = tf.constant([['a', 'y', 'b'],
              ['b', 'y', 'a'],
              ['a', 'y', 'c'],
              ['c', 'y', 'a'],
              ['a', 'y', 'd'],
              ['c', 'y', 'd'],
              ['b', 'y', 'c'],
              ['f', 'y', 'e']])

with tf.Session() as sess:
  for i in X[:,0].eval():
    element= i.decode("utf-8") 
    print(element)

  # Or using sess.run()
  #for j in sess.run(X[:,0]):
  #  element= j.decode("utf-8") 
  #  print(element)

輸出：

a
b
a
c
a
c
b
f