Imported from https://github.com/hiyouga/PBAN-PyTorch