search

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

Spatial Attention mechanism with MinkowskiEngine #489

Open yuan-zm opened 2 years ago

yuan-zm commented 2 years ago

Hello, thanks for your great work.

Can we implement spatial attention mechanism for each input point with MinkowskiEngine?