search

pytorch / nestedtensor

[Prototype] Tools for the concurrent manipulation of variably sized Tensors.
BSD 3-Clause "New" or "Revised" License
252 stars 28 forks source link

Restrict nested dim to one - tests and constructor #413

Closed cpuhrsch closed 3 years ago

cpuhrsch commented 3 years ago

In context of focus on inference this restricts various use cases, but also makes it easier to write efficient kernels.