Pytorch Pairwise cosine similarity
Computes the cosine similarity between all vectors in the input, pytorch originally only computes this between two vectors at a time which can be time consuming
when dealing with thousands of vectors, something that is the case for embedding learning tasks.
-
Notifications
You must be signed in to change notification settings - Fork 0
Simple function that computes pairwaise cosine distance between several vectors at once, pytorch can only compute beween two vectors at a time, which is time consuming and inneficient when you have multiple vectors.
dalisson/pairwise_cosine_distance_pytorch
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Simple function that computes pairwaise cosine distance between several vectors at once, pytorch can only compute beween two vectors at a time, which is time consuming and inneficient when you have multiple vectors.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published