У меня есть тензор (batch, num_points, 2), который состоит из точек в 2D.Я хочу применить кластеризацию k-средних или GMM к этим точкам в пакете и генерировать кластеры k, где заранее определено k.То есть, у меня будут кластерные центроиды формы (batch, k, 2), а также метки кластера для всех точек.
Я заметил, что в TensorFlow есть реализации как для k-средних, так и для GMM-кластеризации, но обе онидля одной партии, и GMM, кажется, глючит.Я хочу применить один из этих алгоритмов к набору данных и хочу, чтобы реализация была дифференцируемой.(Например, я думаю, что выполнение алгоритма EM до 5 раз сделает вычислительный граф дифференцируемым.)
Кто-нибудь может мне порекомендовать реализацию?