axial-positional-embedding (conda): Axial Positional Embedding for Pytorch
Keywords: artificial-intelligence|deep-learning|pytorch