TensorTwoD.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2015 - 2024 Intel Corporation
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef TENSORTWOD_HPP_INCLUDE
#define TENSORTWOD_HPP_INCLUDE
 
#include <cstdint>
#include <memory>
#include <vector>
 
#include "TensorOneD.hpp"
 
namespace geopm
{
    class TensorMath;
 
    class TensorTwoD
    {
        public:
            TensorTwoD();
            TensorTwoD(size_t rows, size_t cols);
            TensorTwoD(size_t rows, size_t cols, std::shared_ptr<TensorMath> math);
            TensorTwoD(const TensorTwoD &other);
 
            TensorTwoD(TensorTwoD &&other);
            TensorTwoD(const std::vector<std::vector<double> > &input);
            TensorTwoD(const std::vector<std::vector<double> > &input,
                       std::shared_ptr<TensorMath> math);
            TensorTwoD(const std::vector<TensorOneD> &input);
            TensorTwoD(const std::vector<TensorOneD> &input, std::shared_ptr<TensorMath> math);
            size_t get_rows() const;
            size_t get_cols() const;
            void set_dim(size_t rows, size_t cols);
            TensorOneD operator*(const TensorOneD &) const;
            TensorOneD &operator[](size_t idx);
            TensorOneD operator[](size_t idx) const;
            TensorTwoD& operator=(const TensorTwoD &other);
 
            TensorTwoD& operator=(TensorTwoD &&other);
            //         data
            bool operator==(const TensorTwoD &other) const;
 
            const std::vector<TensorOneD> &get_data() const;
 
            void set_data(const std::vector<TensorOneD> &);
 
        virtual ~TensorTwoD() = default;
        private:
            std::vector<TensorOneD> m_mat;
            std::shared_ptr<TensorMath> m_math;
    };
}
#endif