After some search work, I found that there is a type specially designed for holding the size of an axis: `dim_t`. I will try changing it to this type. `index_t` is not quite suitable since as far as I know it is unsigned, but `k` can be negative in our case.
[ Full content available at: https://github.com/apache/incubator-mxnet/pull/12430 ] This message was relayed via gitbox.apache.org for [email protected]
