Function edges_slice

examples/image_processing/brain_segmentation.cpp:27–33

//};

array edges_slice(array x) {
    array ret;
    static array kernelx = array(dim4(3, 3), h_sx_kernel);
    static array kernely = array(dim4(3, 3), h_sy_kernel);
    ret                  = convolve(x, kernelx) + convolve(x, kernely);
    return abs(ret);
}

array gauss(array x, float u, float s) {
    double f = 1 / sqrt(2 * af::Pi * s * s);