How to index a 4D tensor in your own cuda kernel function

04 Jan 2022

See this study note How to index a 4D tensor in your own cuda kernel function.