Function af_diag_extract

src/api/c/data.cpp:310–349

}

af_err af_diag_extract(af_array *out, const af_array in, const int num) {
    try {
        const ArrayInfo &in_info = getInfo(in);
        af_dtype type            = in_info.getType();

        if (in_info.ndims() == 0) {
            return af_create_handle(out, 0, nullptr, type);
        }

        DIM_ASSERT(1, in_info.ndims() >= 2);

        af_array result = nullptr;
        switch (type) {
            case f32: result = diagExtract<float>(in, num); break;
            case c32: result = diagExtract<cfloat>(in, num); break;
            case f64: result = diagExtract<double>(in, num); break;
            case c64: result = diagExtract<cdouble>(in, num); break;
            case s32: result = diagExtract<int>(in, num); break;
            case u32: result = diagExtract<uint>(in, num); break;
            case s64: result = diagExtract<intl>(in, num); break;
            case u64: result = diagExtract<uintl>(in, num); break;
            case s16: result = diagExtract<short>(in, num); break;
            case u16: result = diagExtract<ushort>(in, num); break;
            case s8: result = diagExtract<schar>(in, num); break;
            case u8:
                result = diagExtract<uchar>(in, num);
                break;
                // Removed because of bool type. Functions implementations
                // exist.
            case b8: result = diagExtract<char>(in, num); break;
            case f16: result = diagExtract<half>(in, num); break;
            default: TYPE_ERROR(1, type);
        }

        std::swap(*out, result);
    }
    CATCHALL;

    return AF_SUCCESS;
}

template<typename T>
inline af_array triangle(const af_array in, const bool is_upper,