У меня есть массив 130 000 x 1024 (число с плавающей запятой 64), мне нужно вычислить косинусное сходство между всеми векторами так, чтобы на выходе было 130 k x 130 k. Я пытался использовать sklearn.metrics.pairwise сходство косинусов и Pytorch's. В обоих методах я достигаю переполнения процессора ..
Есть идеи, как это сделать?