Task Description

Implement Vertex alignment operation as described in https://pytorch3d.readthedocs.io/en/latest/_modules/pytorch3d/ops/vert_align.html

Write your code and tests in this folder: https://github.com/PurdueDualityLab/tf-models/tree/mesh_rcnn/official/vision/beta/projects/mesh_rcnn/ops

🚨 Please follow the Google Style Guide 🚨

Before opening a pull request:

• Differentially tested against original implementation ✅
• All test cases are written and passed 👍
• Code is linted ✏️