Function rearrange_image

examples/pde/bhrt.cpp:893–899

* @brief Transform the array of pixels to the correct image format to display * * @param image * @param width * @param height * @return af::array */

 * @return af::array
 */
af::array rearrange_image(const af::array& image, uint32_t width,
                          uint32_t height) {
    return af::clamp(af::moddims(image, af::dim4(width, height, 3)).T(), 0.0,
                     255.0)
               .as(f32) /
           255.f;
}

/**
 * @brief Returns an rgb image containing the raytraced black hole from the