请问下，论文中figure2的mask示例图，是在训、推过程中都使用吗，相关的代码在哪？（我通过训练代码发现，训练是将两个由lens_to_mask生成的下三角mask concat的，感觉找的地方不对）