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

在tensorflow中有一个二维张量

比如说一个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（）函数非常复杂，无法通过矢量化直接使用

---

非常感谢

也许您正在寻找的是Tensorflow中的函数

map\u fn（a，elems）

将一个张量，

elems

沿其第一维解压为一系列切片，然后将提供的函数

a

应用于每个切片，然后通过沿第一维串联将输出再次组合为单个张量

听起来你想要的是

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

---

其中，

X

是您提供的张量。

凤凰城，这似乎不同于仅使用reduce\u平均值。