Is it able to count also for layers from torch_geometric?

[harborsarah]

Dear authors,

Since my model contains also layers from torch_geometric, and my model has multiple inputs. Some of them have a predefined size but one input is the point cloud, which does not have a predefined shape from frame to frame. Instead, it is generated via torch_geometric.data.Batch.from_data_list() etc.

I was woundering, if this package can also handel this?

Thanks a lot for your reply.