我有一个tensor数据点points，其形状为(1, #data, #dimension)，另一个为dist，其形状为(#cluster, 1, #dimension)，最后一个是cluster，形状为(#data, )，表示points所属的群集索引。现在我想计算每个集群的平均值。所以预期的输出应该有(#cluster, 1, #dimension)的形状。我做了tf.gather(points, cluster)之类的事情，得到的输出形状为(#data, #data, #dimension)。现在我不知道应该怎么处理它。你能帮我吗？
编辑：
例如，points为[[2, 3], [1, 4], [8, 10]]，cluster为[1, 0, 2]，预期输出应为[[[0,0], [2, 3], [0, 0]], [[1, 4], [0, 0], [0, 0]], [[0, 0], [0, 0], [8, 10]]]，因此方便计算的意思。