Allow batch dimensions in dot and cosine similarity

hyperdimensional-computing / torchhd

Torchhd is a Python library for Hyperdimensional Computing and Vector Symbolic Architectures

https://torchhd.readthedocs.io

MIT License

229 stars 24 forks source link

Allow batch dimensions in dot and cosine similarity #119

Closed mikeheddes closed 1 year ago

mikeheddes commented 1 year ago

The current documentation states that the dot and cosine similarity accept batch dimensions but this was never actually implemented. This PR allows batch dimension to be used in dot and cosine similarity.