Function diagExtract

src/backend/cpu/diagonal.cpp:42–50

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

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

    return out;
}

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