matrix.hpp

Go to the documentation of this file.

#pragma once
 
#include <initializer_list>
#include <iostream>
#include <tuple>
#include <xtensor/containers/xarray.hpp>
#include <xtensor/core/xmath.hpp>
#include <xtensor/generators/xrandom.hpp>
#include <xtensor/io/xio.hpp>
#include <xtensor/reducers/xreducer.hpp>
#include <xtensor/views/xview.hpp>
 
namespace utils {
    template<typename T>
    class Matrix {
    public:
        Matrix() : rows_(0), cols_(0) {}
 
        Matrix(size_t rows, size_t cols);
 
 
        Matrix(size_t rows, size_t cols, T value);
 
        Matrix(std::initializer_list<std::initializer_list<T>> list);
 
        template<typename U>
        friend std::ostream &operator<<(std::ostream &os, const Matrix<U> &matrix);
        template<typename U>
        friend Matrix<U> dot(const Matrix<U> &a, const Matrix<U> &b);
        template<typename U>
        friend U sum(const Matrix<U> &matrix);
        template<typename U>
        friend U mean(const Matrix<U> &matrix);
 
        T &operator()(size_t row, size_t col);
 
        const T &operator()(size_t row, size_t col) const;
 
        Matrix operator+(const Matrix &other) const;
 
        Matrix operator-(const Matrix &other) const;
 
        Matrix operator*(const Matrix &other) const;
 
        Matrix transpose() const;
 
        Matrix reshape(size_t new_rows, size_t new_cols) const;
 
        T determinant() const;
 
        Matrix inverse() const;
 
        auto eigenvalues() const;
 
        [[nodiscard]] size_t rows() const { return rows_; }
 
        [[nodiscard]] size_t cols() const { return cols_; }
 
        [[nodiscard]] size_t size() const { return rows_ * cols_; }
 
        [[nodiscard]] std::tuple<size_t, size_t> shape() const { return {rows_, cols_}; }
 
        static Matrix zeros(size_t rows, size_t cols);
 
        static Matrix ones(size_t rows, size_t cols);
 
        static Matrix identity(size_t size);
 
        static Matrix random(size_t rows, size_t cols, T min, T max);
 
        xt::xarray<T> &data() { return data_; }
 
        const xt::xarray<T> &data() const { return data_; }
 
    private:
        xt::xarray<T> data_;
 
        size_t rows_, cols_;
    };
 
    template<typename T>
    std::ostream &operator<<(std::ostream &os, const Matrix<T> &matrix);
 
    template<typename T>
    Matrix<T> dot(const Matrix<T> &a, const Matrix<T> &b);
 
    template<typename T>
    T sum(const Matrix<T> &matrix);
 
    template<typename T>
    T mean(const Matrix<T> &matrix);
 
    using MatrixF = Matrix<float>;
 
    using MatrixD = Matrix<double>;
 
} // namespace utils