在张量流中寻找二维张量分组平均值?

时间:2018-07-08 02:21:08

标签: python tensorflow

我想在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。我需要将该操作推广到任意AL。我可以为Numpy构建操作,但找不到Tensorflow的任何方法。

0 个答案:

没有答案