Function diagExtract

src/backend/opencl/diagonal.cpp:34–42

template<typename T>
Array<T> diagExtract(const Array<T> &in, const int num) {
    const dim_t *idims = in.dims().get();
    dim_t size         = std::min(idims[0], idims[1]) - std::abs(num);
    Array<T> out       = createEmptyArray<T>(dim4(size, 1, idims[2], idims[3]));

    kernel::diagExtract<T>(out, in, num);

    return out;
}

#define INSTANTIATE_DIAGONAL(T)                                          \
    template Array<T> diagExtract<T>(const Array<T> &in, const int num); \