Tan_pytorch_segmentation/pytorch_segmentation/PV_Attention/ACmix-Attention.py

from model.attention.ACmix import ACmix
import torch

if __name__ == '__main__':
    input = torch.randn(50, 256, 7, 7)
    acmix = ACmix(in_planes=256, out_planes=256)
    output = acmix(input)
    print(output.shape)