Calculate a multi-dimensional coordinates' linearized index
| 522 | |
| 523 | // Calculate a multi-dimensional coordinates' linearized index |
| 524 | dim_t ravelIdx(af::dim4 coords, af::dim4 strides) { |
| 525 | return std::inner_product(coords.get(), coords.get() + 4, strides.get(), |
| 526 | 0LL); |
| 527 | } |
| 528 | |
| 529 | // Calculate a linearized index's multi-dimensonal coordinates in an |
| 530 | // af::array, |