我有一系列功能，代表某些类的对象。该数组中的每个条目都分配了一个表示类的标签。看起来可能像这样（由标签1和2表示的3行和两个类的数组）：
[[ 0.1, 0.2, 0.4, 0.5. | 1]
[-0.2, 0.4, -0.3, 0.2 | 2]
[-0.2, -0.9, -0.7, 0.4 | 1]]
我正在努力的是在张量流中有效地计算上述特征所表示的簇的中心。使用numpy索引，我会得到这样的东西：
import numpy_indexed as npi
npi.group_by(...).mean(...)
现在我需要对tensorflow进行同样的操作。