nestedtensor
nestedtensor copied to clipboard
pytorch
[Prototype] Tools for the concurrent manipulation of variably sized Tensors.
See pytorch/pytorch#26817 for what this does FB only: [Chronos job page](https://www.internalfb.com/intern/chronos/job/?jobname=pytorch_nestedtensor_nightly_trigger&smc=chronos_gp_admin_client)
## 🐛 Bug shape, size, dim, nested_size, nested_dim, tensor_dim, etc don't work. ## To Reproduce Steps to reproduce the behaviour: initialise nestedtensor nt and call nt.shape ``` >>> import torch...
## 🚀 Feature Support slice operation on "ragged" dimension ## Motivation In preproc we often wants to operates over variable-width list, such as token ids in text domain, or sparse...
## 🐛 Bug Can index into a NestedTensor without grad, but cannot index into a NestedTensor with grad ## To Reproduce Steps to reproduce the behavior: ``` import torch c...
aten reference: [permute](https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/native/native_functions.yaml#L2230). Semantics here are limited if we're to maintain the view functionality of pytorch. A user may permute tensor dimensions xor nestedtensor dimensions. Permutations at a per-tensor dimension...
Summary: Fixes missing kernel guards as identified by D30072495 Differential Revision: D31553158
Say the original data is `A=[tensor([1., 2., 3.]), tensor([2.]), tensor([1., 2., 3., 4.])]`. For convenience of some other operations, I concat the data into `A_cat = tensor([1., 2., 3., 2.,...
### The main reason I want to use nestedtensor and what value I want it to add * Building a time series modelling system with pytorch where each step may...
