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