Method AsDiagonalOp

extensions/include/cuMat/src/UnaryOps.h:381–387

public:
    explicit AsDiagonalOp(const MatrixBase<_Child>& child)
        : child_(child.derived())
        , size_(child.rows()==1 ? child.cols() : child.rows())
    {
        CUMAT_STATIC_ASSERT(internal::traits<_Child>::RowsAtCompileTime == 1 || internal::traits<_Child>::ColsAtCompileTime == 1,
            "The child expression must be a compile-time row or column vector");
    }

    __host__ __device__ CUMAT_STRONG_INLINE Index rows() const { return size_; }
    __host__ __device__ CUMAT_STRONG_INLINE Index cols() const { return size_; }