Does pyg temporal support multidimensional edge_attr as pyg?

benedekrozemberczki / pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

MIT License

2.67k stars 377 forks source link

Does pyg temporal support multidimensional edge_attr as pyg? #245

Open Sanjaykrishnamurthy opened 1 year ago

Sanjaykrishnamurthy commented 1 year ago

Hi benedekrozemberczki,

Thanks for the fantastic library. For my usecase I have a set of edge_attributes. I tried giving edge_attr as a vector, but it throws error.

So jus wanted to know if it support multidim edge features.

lucianocheng commented 2 months ago

@Sanjaykrishnamurthy what is the error you are seeing? And which iterator are you using?

DynamicGraphTemporalSignal passes edge_weight directly into edge_attr of torch_geometric.data.Data, which should support multi-dimensional tensors.