У меня есть массив функций, представляющих некоторые классы объектов.Каждой записи в этом массиве назначена метка, которая выражает класс.Это может выглядеть примерно так (массив из 3 строк и двух классов, представленных метками 1 и 2):
[[ 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(...)
Теперь мне нужно сделать то же самое с тензорным потоком.