search

the-lay / tiler

N-dimensional NumPy array tiling and merging with overlapping, padding and tapering
https://the-lay.github.io/tiler/
MIT License
65 stars 10 forks source link

[IDEA] Tiling PyTorch tensors #14

Open the-lay opened 2 years ago

the-lay commented 2 years ago

Is your feature request related to a problem? Please describe. Pytorch's API is very numpy-like, but has differences in underlying memory model. It might be interesting to take a look what's happening there and whether it is possible to tile pytorch tensors too.

alisterburt commented 2 years ago

I've run into issues with this myself and have needed to take things off the gpu just to tile which isn't ideal - I'll take a look and see what the sticking points are