Closed aurorarossi closed 1 year ago
A few comments:
[embed_dim, seq_len, batch_dim]
, so here we probably want to have [embed_dim, t, num_nodes]
and [embed_dim, t, num_edges]
.TemporalDataStore
structure at all? Can't we just use DataStore
, since it can store arbitrary dimension arrays already? We could extend it to optionally check the size of the second to last dimension as well.
This PR implements TemporalDataStore.