Giters

oelin / axial-attention

An implementation of axial attention in PyTorch.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

axial-attention

An implementation of axial attention in PyTorch.

About

An implementation of axial attention in PyTorch.

License:MIT License

Languages

Language:Python 100.0%