I have a batch of polygons which could be a NumPy array, torch tensor, or any other nd-array of shape (230_000, 3, 2). 230_000 is the number of polygons, 3 indicates it’s a triangle, 2 is its x and y coordinates. I also have a batch of features of shape (230_000, 3, 3) (last is the color), it’s best the