bunsen 0.0.1

burn neural network extension library
Documentation
1
2
3
4
5
6
7
8
9
10
//! Window attention operations for Swin Transformer v2.
mod attention;
mod attention_mask;
mod pos_bias;
mod pos_grid;

pub use attention::*;
pub use attention_mask::*;
pub use pos_bias::*;
pub use pos_grid::*;