如何在Tensorflow中迭代張量？

Question

假設我們得到像這樣的張量x = [[[1,2],[3,4]],[[5,6],[7,8]]] 。 我想要一個tensorflow操作，使得它返回第一個嵌套張量的每個元素的加1。 即結果操作將返回[[[2,3],[3,4]],[[6,7],[7,8]]] 。 我知道tf.map_fn操作，但我找不到使用此操作實現的方法。 如何解決這個問題呢？

Answer 1

您可以在第一個軸上分割張量，在第一個軸上添加一個張量，然后使用tf.stack堆疊新舊張量：

>>> x = tf.constant([[[1, 2], [3, 4]],[[5, 6], [7, 8]]])
>>> with tf.Session() as sess:
...    sess.run(tf.stack((x[:,0] + 1, x[:, 1]), axis=1))
... 
array([[[2, 3],
        [3, 4]],

       [[6, 7],
        [7, 8]]], dtype=int32)