使用gather_nd在循环中获取结果-所有输入的形状必须匹配答案

【问题标题】：use gather_nd to acquire result in a loop - shapes of all inputs must match使用gather_nd在循环中获取结果-所有输入的形状必须匹配
【发布时间】：2019-06-28 11:16:41
【问题描述】：

我在 numpy 中有这个例子：

import numpy as np
import tensorflow as tf

a = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9],
              [10, 11 , 12],
              [13, 14, 15]])

res = np.zeros((5, 2), dtype=object)

for idx in range(0, len(a)-2, 2):
    a0 = a[idx]
    a1 = a[idx + 1]
    a2 = a[idx + 2]
    c = a0 + a1 + a2

    res[idx:idx + 2] = ([idx, c])

res

array([[0, array([12, 15, 18])],
       [0, array([12, 15, 18])],
       [2, array([30, 33, 36])],
       [2, array([30, 33, 36])],
       [0, 0]], dtype=object)

我想在 tensorflow 中做：

a_tf = tf.convert_to_tensor(a)
res_tf = tf.zeros((5, 2), dtype=object)

for idx in range(0, a.shape[0]-2, 2):
    a0 = tf.gather_nd(a, [idx])
    a1 = tf.gather_nd(a, [idx + 1])
    a2 = tf.gather_nd(a, [idx + 2])
    c = a0 + a1 + a2

    res = tf.gather_nd([idx, c], [idx:idx +2])

直到与c 的计算一致。

最后一行 (res) 它给了我：

res = tf.gather_nd([idx, c], [idx:idx +2])
                                     ^
SyntaxError: invalid syntax

我不确定如何收到结果。

更新

基本上，问题在于[idx, c] 是列表类型并试图做：tf.convert_to_tensor([idx, c]，给出：

InvalidArgumentError: Shapes of all inputs must match: values[0].shape = [] != values[1].shape = [3] [Op:Pack] name: packed/

【问题讨论】：

标签： python-3.x tensorflow tensorflow2.0

【解决方案1】：

res = tf.gather_nd([idx, c], [idx:idx +2])

语法不正确。如果要提取索引，应该是

res = tf.gather_nd([idx, c], range(idx, idx +2))

后者也可能会引发错误。 range(idx, idx +2) 中的索引高于列表[idx, c] 的索引。

此外，除非使用ragged tensors，否则无法创建形状为res 的张量。这是您尝试做的可能的解决方法

a_tf = tf.convert_to_tensor(a)
res_tf = tf.zeros((5, 2), dtype=object)
l = []
for idx in range(0, a.shape[0]-2, 2):
    a0 = tf.gather_nd(a, [idx])
    a1 = tf.gather_nd(a, [idx + 1])
    a2 = tf.gather_nd(a, [idx + 2])
    c = a0 + a1 + a2
    helper = [idx]
    helper.extend(c.numpy().tolist())
    l.append(helper)

print(tf.constant(l))

【讨论】：