huggingface / huggingface/datasets
Use `torch.nested_tensor` for arrays of varying length in torch formatter
Open
enhancement
- Dominant language
- Python
- Stars
- 22k
- Forks
- 3.4k
- Avg merge
- 5d 7h
- Merged PRs (30d)
- 17
Description
Use `torch.nested_tensor` for arrays of varying length in `TorchFormatter`.
The PyTorch API of nested tensors is in the prototype stage, so wait for it to become more mature.
Contributor guide
Assessment
This issue has not been assessed yet.