Method coeff

extensions/include/cuMat/src/UnaryOps.h:482–486

    __host__ __device__ CUMAT_STRONG_INLINE Index batches() const { return child_.batches(); }

    __device__ CUMAT_STRONG_INLINE Scalar coeff(Index row, Index col, Index batch, Index index) const
    {
        using Functor = internal::ExtractDiagonalFunctor<MatrixType>;
        return Functor::extractDiagonal(child_.derived().coeff(row, row, batch, index));
    }
};