nestedtensor icon indicating copy to clipboard operation
nestedtensor copied to clipboard

[Prototype] Tools for the concurrent manipulation of variably sized Tensors.

Results 49 nestedtensor issues
Sort by recently updated
recently updated
newest added

When I want to train model with nestedtensor, it raise this error, how can I solve it?

I want to use torch.gather for NestedTensor, but report this error **RuntimeError: Could not run 'aten::gather' with arguments from the 'PrivateUse1' backend.** How can I solve it?

Our old version of as_nested_tensor allowed users to create NestedTensor as a container that wraps lists of Tensors and allows modifications to trickle down to the constituents. However, [this was...

There is no easy way to find out the number of tensors in Nested Tensor

enhancement

Repro: `a = creation.nested_tensor([], dtype=torch.int64)`

bug

the required manual workaround in test_unbind_dim

bug
testing

For example [conv2d](https://github.com/pytorch/nestedtensor/blob/master/nestedtensor/csrc/functions.h#L200) doesn't align with [aten's definition](https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/native/Convolution.cpp#L504). I suspect the same is true for others.

bug

We should write up a careful comparison of the advantages and disadvantages of writing NestedTensor as a Tensor-esque type as it is currently done or as a subclass of [TensorImpl](https://github.com/pytorch/pytorch/blob/f9f135c5d8fc1986e112e62473b3498cdf4d1c43/c10/core/TensorImpl.h).

design

e.g. misaligned dtype in input or not a view of a buffer during C++ construction

enhancement