Is the input to the model that is point cloud data supported?

[auniquesun]

@zhijian-liu As the title shows, I want to count the #params of a point cloud model (e.g., PointBERT) instead of an image model, the example input of point cloud data is of shape (batch_size, 1024, 3). It means each point cloud has 1024 points with 3 coordinates, can I achieve this goal by calling the API in you library?

Thanks.