axial_positional_embedding

Depended on by

• dalle-pytorch

• nuwa-pytorch