| 30 | |
| 31 | template<typename Ti, typename Tw, typename To> |
| 32 | Array<To> mean(const Array<Ti> &in, const int dim) { |
| 33 | dim4 odims = in.dims(); |
| 34 | odims[dim] = 1; |
| 35 | Array<To> out = createEmptyArray<To>(odims); |
| 36 | static const mean_dim_func<Ti, Tw, To> mean_funcs[] = { |
| 37 | kernel::mean_dim<Ti, Tw, To, 1>(), kernel::mean_dim<Ti, Tw, To, 2>(), |
| 38 | kernel::mean_dim<Ti, Tw, To, 3>(), kernel::mean_dim<Ti, Tw, To, 4>()}; |
| 39 | |
| 40 | getQueue().enqueue(mean_funcs[in.ndims() - 1], out, 0, in, 0, dim); |
| 41 | return out; |
| 42 | } |
| 43 | |
| 44 | template<typename T, typename Tw> |
| 45 | using mean_weighted_dim_func = |