如何在张量流中循环遍历张量中的每一行

Question

我在张量流中有一个二维张量，

比如说一个 2*4 张量 [[1.,2.,3.,4.],[2.,4.,5.,6.]]。

我有一个函数 a() 让张量中的每一行都通过，然后对 a() 的所有结果求和。怎么做（不在会话中做）？

输出应该是 a([1.,2.,3.,4.]) + a([2.,4.,5.,6.])，实际上我有一个非常大的张量很多行。

这和reduce_sum不同，因为这里的a()函数比较复杂，不能通过向量化直接使用。

非常感谢！

Answer 1

也许您正在寻找的是 Tensorflow 中的 map_fn 函数。 map_fn(a, elems) 将张量 elems 沿其第一个维度解包为切片序列，然后将提供的函数 a 应用于每个切片，然后通过沿第一个维度连接将输出再次组合成单个张量。

听起来你想要的是

Y = map_fn(a, X)
answer = reduce_sum(Y, axis=0)

X 是您提供的张量。