nestedtensor
nestedtensor copied to clipboard
[Prototype] Tools for the concurrent manipulation of variably sized Tensors.
There should be a way to replace tensors inside of an NT via assigning.
``` a = nt.nested_tensor([ nt.nested_tensor([]), nt.nested_tensor([ torch.tensor([1]) ]) ]) ``` same for cuda should succeed ``` a = nt.nested_tensor([ [], [ torch.tensor([1]) ] ]) ```
Repro: `creation.nested_tensor([]).is_contiguous() != creation.nested_tensor([], dtype=torch.float).is_contiguous()`
I found out that `tensorwise` actually just runs a for loop over the nested tensors. I benchmarked `tensorwise` against `map`, list comprehension and for loop. (Un)surprisingly, `tensorwise` performs much slower...
I got a pickler error in multiprocessing that it cant pickle the nestedtensor object. does it mean i cant use nestedtensor for Dataset? I guess using nestedtensor in Dataset is...
As of right now, we are not supporting empty tensors in nested tensor when calling `to_tensor_mask()` method. Example case: ``` nt1 = nt.nested_tensor([ nt.nested_tensor([ nt.nested_tensor([ torch.tensor([], dtype=torch.float) ]), nt.nested_tensor([ torch.tensor([1],...