.size()里面的维度参数是不是错了，应该是.size(0)

[wenxiang2001]

https://github.com/wenjtop/transformer/blame/bb7fe7e2d2bf873440af2c80c28f7ffa1d5b93cd/transformer.py#L24