A simple implement of Boundary Attention https://arxiv.org/abs/2401.00935, with 207,516 learnable parameters.

I am not sure wheather Gather and Slice operators are correct, and many training details mentioned in paper are not clear yet.