/*------------------------------------------------------------------------------
 *	Copyright (c) 2023 by Bai Bing (seread@163.com)
 *	S++ COPYING file for copying and redistribution conditions.
 *
 *  Alians IT Studio.
 *----------------------------------------------------------------------------*/
#pragma once

#include <string>

#include "ASMatrix.h"

namespace ais
{
    //============================================================================
    // Method Description:
    /// Return specified diagonals.
    ///
    /// @param array
    /// @param offset (Defaults to 0)
    /// @param axis (Optional, default ROW) axis the offset is applied to
    /// @return Matrix
    ///
    template <typename dtype>
    Matrix<dtype> diagonal(const Matrix<dtype> &array, int64_t offset = 0, Axis axis = Axis::ROW)
    {
        return array.diagonal(offset, axis);
    }
} // namespace ais