How can I do matrix multiplication based on MinkowskiEngine?

NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

https://nvidia.github.io/MinkowskiEngine

Other

2.43k stars 360 forks source link

How can I do matrix multiplication based on MinkowskiEngine? #582

Open xing-yuu opened 6 months ago

xing-yuu commented 6 months ago

I have a matrix A with shape 1,1,3,5, and another matrix B with shape [1,1,5,4]. Now, using function MinkowskiEngine.MinkowskiOps.to_sparse, I want to convert them into sparse matrices and multiply the two matrices to obtain a matrix with shape [1,1,3,4].

How can I do？