Я построил SparseTensors и в pytorch, и в tenorflow, и я ищу способ получить расхождение KL между строками или столбцами (матрица симметрична).
В частности, матрица является матрицей смежности и требует, чтобы она была SparseTensor в тензорной форме, поскольку при преобразовании .todense() она становится 400 Гбит
Существует ли способ получить расхождение KL между каждой строкой, аналогично операции, выполняемой при получении матрицы расстояний путем применения метрики расстояния для каждой строки?
Я посмотрел на документацию SparseTensor для pytorch и тензорного потока, но их, похоже, не хватало.Я открыт для других пакетов, которые используют графический процессор на разреженных матрицах или тензорах.