大佬，“手撕transformer decoder部分的inference过程，重点看beam search的实现”，直接写beam search是不是有点太长了，感觉好难，然后这块如果要看的话，是直接看google transformer的tensorflow源码吗？