对联
基于中的变压器实现自动对对联。贯通的放置有:更改了DecoderLayer中第二层注意的查询,键,值以捕获上联的模式和下联的语义,增加了一倍层内存的线性叠加以捕获上联的语义,调整了PositionalEncoding的方式。另外还实现了greedy,一维波束搜索,二维波束搜索等解码方式。
输出示例
上联
参考下联
系统输出
锦瑟无弦惊日月
铜钱有脚走街衢
瑶琴有韵醉春秋
金龙辞旧,秦岭山河千般秀
银蛇迎春,神州日月万象新
银蛇迎新,神州大地万里春
天涯苦远,讯信迷茫,唯凭一纸传吟,纵有相思