torch-bmv

Batched matrix-vector multiplication in PyTorch which intelligently rearranges tensor shapes for efficient computation.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install torch-bmv

Releases

Version Released Bullseye
Python 3.9
Bookworm
Python 3.11
Files
0.1.1 2025-07-01

Issues with this package?

Page last updated 2025-07-01 23:04:21 UTC