生成式：PolyGen: An Autoregressive Generative Model of 3D Meshes【vertex_model.py image标签情况下代码解读】

这份文章只介绍从forward开始，数据经历了哪些操作，分别对应的意义是什么。在另外一篇文章中，会将这这篇文章对应的代码中的smaple进行详细讲解，以理解逐点生成的过程。

文章对应代码仓库为：https://github.com/anshulcgm/polygen中的vertex_model.py。

话不多说，从forward开始进行解析。

以单纯的vertex数据为例，不涉及Image.
注意查看每个函数的操作流程，以及数据维度的变换。

1、forward

    def forward(self, batch: Dict[str, torch