Machine learning 在GPU中计算余弦相似性的最快方法

Machine learning 在GPU中计算余弦相似性的最快方法,machine-learning,gpu,google-colaboratory,recommendation-engine,cosine-similarity,Machine Learning,Gpu,Google Colaboratory,Recommendation Engine,Cosine Similarity,所以我有一个巨大的tfidf矩阵,有超过一百万条记录,我想找出这个矩阵与自身的余弦相似性。我正在使用colab来运行代码,但我不确定如何最好地利用colab提供的gpu 顺序运行代码- tfidf_矩阵=tf.fit_变换(df['categories']) 余弦相似性=线性核(矩阵,矩阵) 我们有没有办法使用jit或其他方式来并行化代码

所以我有一个巨大的tfidf矩阵,有超过一百万条记录,我想找出这个矩阵与自身的余弦相似性。我正在使用colab来运行代码,但我不确定如何最好地利用colab提供的gpu

顺序运行代码-

tfidf_矩阵=tf.fit_变换(df['categories'])

余弦相似性=线性核(矩阵,矩阵)

我们有没有办法使用jit或其他方式来并行化代码