Specialized mask kernel for to_mask(dim=2) - Githubissues

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

Specialized mask kernel for to_mask(dim=2) #466

Closed cpuhrsch closed 2 years ago