NestedTensor cannot be pickled #104198
Description
import torch
import pickle
a, b = torch.arange(3), torch.arange(5) + 3
nt = torch.nested.nested_tensor([a, b])
with open('tensor.pkl', 'wb') as f:
pickle.dump(nt, f)click to see original report
🚀 The feature, motivation and pitch
Hello,
I am trying to make a custom dataset class of ZINC dataset on PyTorch geometric with nested tensors using "torch.nested.nested_tensor", and when executing I got the following error:
"Internal error: NestedTensorImpl doesn't support sizes."
Is there a way to get the of the different tensors in the nested tensor?
Thanks
Alternatives
No response
Additional context
No response
cc @mruberry @mikaylagawarecki @cpuhrsch @jbschlosser @bhosmer @drisspg