python - 在张量流中找到二维张量分组的平均值？

Question

我想在 Tensorflow 中找到 2D 张量的按组逐行均值。我将通过一个例子展示我想要实现的目标。例如，如果有一个张量A=[[0,1],[1,3],[2,6],[3,1],[2,7]]和一个显示组的列表列表L=[[0,1],[3],[2,4]]。所以在 list 中L，有 3 个列表，所以会有 3 个组。所以输出应该是一个 3 X 2 矩阵，第一行是 0,1A行的平均值，第二行是 3A行的平均值，第三行是 2,4 行的平均值A。我需要将操作推广到任意任意A, L。我可以为 Numpy 构建一个操作，但我找不到任何用于 Tensorflow 的方法。

标签： pythontensorflow