Imported from https://github.com/kaituoxu/Speech-Transformer