| 34 | |
| 35 | template<typename T> |
| 36 | void gaussian1D(T* out, int const dim, double sigma = 0.0) { |
| 37 | if (!(sigma > 0)) sigma = 0.25 * dim; |
| 38 | |
| 39 | T sum = (T)0; |
| 40 | for (int i = 0; i < dim; i++) { |
| 41 | int x = i - (dim - 1) / 2; |
| 42 | T el = 1. / std::sqrt(2 * af::Pi * sigma * sigma) * |
| 43 | std::exp(-((x * x) / (2 * (sigma * sigma)))); |
| 44 | out[i] = el; |
| 45 | sum += el; |
| 46 | } |
| 47 | |
| 48 | for (int k = 0; k < dim; k++) out[k] /= sum; |
| 49 | } |
| 50 | } // namespace cpu |
| 51 | } // namespace arrayfire |