Function diag

src/api/cpp/data.cpp:191–200

}

array diag(const array &in, const int num, const bool extract) {
    af_array res;
    if (extract) {
        AF_THROW(af_diag_extract(&res, in.get(), num));
    } else {
        AF_THROW(af_diag_create(&res, in.get(), num));
    }

    return array(res);
}

array moddims(const array &in, const unsigned ndims, const dim_t *const dims) {
    af_array out = 0;