Tensorflow与python累加器代码等效的代码

Question

我正在尝试实现https://github.com/srl-freiburg/momo/blob/master/python/momo/accum.py中引用的累加器的python代码，但是对于变量赋值，tensorflow似乎不如numpy灵活。他们有什么方法可以处理此代码以使其在tensorflow下工作吗？我感兴趣的唯一函数是np.sum函数。

代码中的主要动作是以下代码段：

vals = np.empty(size, dtype='O')
for s in product(*[range(k) for k in size]):
    vals[s] = []
for s in product(*[range(k) for k in a.shape]):
    indx = tuple(accmap[s])
    val = a[s]
    vals[indx].append(val)

# Create the output array.
out = np.empty(size, dtype=dtype)
for s in product(*[range(k) for k in size]):
    if vals[s] == []:
        out[s] = fill_value
    else:
        out[s] = np.sum(vals[s])
return out

我将不胜感激。

最诚挚的问候， 埃萨