Does pyg support adding new attributes to Batch
objects?
#9126
Unanswered
genesis-jamin
asked this question in
Q&A
Replies: 1 comment
-
In this case, you would also need to add batch['mean_pos'].pos = ...
batch._slice_dict['mean_pos']['pos'] = torch.arange(0, batch['ligand'].max()) There exists some plans to make this easier, see #8414 |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
We're doing something like this in our code right now:
However we find that later when we call
batch.to_data_list()
, we get an error like this:What's the correct way to add new attributes to the
Batch
object?Beta Was this translation helpful? Give feedback.
All reactions