In the cupy docs on user defined kernels (https://docs.cupy.dev/en/stable/user_guide/kernel.html), there is a section defining certain variables that are predefined, like _ind.size()
and i
for things like manual indexing in element-wise kernels.
It seems like these variables are make sense only for 1D arrays; I was wondering how they scale/change for larger multidimensional arrays; explicit references to element-wise manipulation of ndarrays are not really given anywhere (to my knowledge).
1