hlp-ai / mt-data

MT Data
Apache License 2.0
1 stars 2 forks source link

计算向量余弦相似度改用批处理,测试计算单个句对边缘值耗时约为0.19 ms #9

Closed CyberRambler closed 1 year ago

CyberRambler commented 1 year ago

批计算一个向量与其临近的k个向量(组合成二维张量)的余弦相似度。 在48维向量,k=8的条件下,计算单个句对边缘值耗时约为0.19 ms。