TensorOneD.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2015 - 2024 Intel Corporation
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef TENSORONED_HPP_INCLUDE
#define TENSORONED_HPP_INCLUDE
 
#include <cstdio>
#include <memory>
#include <vector>
 
namespace geopm
{
    class TensorMath;
 
    class TensorOneD
    {
        public:
            TensorOneD();
            TensorOneD(size_t n);
            TensorOneD(const TensorOneD &other);
 
            TensorOneD(TensorOneD &&other);
            TensorOneD(const std::vector<double> &input);
 
            TensorOneD(const std::vector<double> &input, std::shared_ptr<TensorMath> math);
 
            void set_dim(size_t dim);
            size_t get_dim() const;
 
            TensorOneD operator+(const TensorOneD &other) const;
            TensorOneD operator-(const TensorOneD &other) const;
            double operator*(const TensorOneD &other) const;
            TensorOneD& operator=(const TensorOneD &other);
 
            TensorOneD& operator=(TensorOneD &&other);
 
            //         data
            bool operator==(const TensorOneD &other) const;
 
            double &operator[](size_t idx);
            double operator[](size_t idx) const;
 
            TensorOneD sigmoid() const;
 
            const std::vector<double> &get_data() const;
 
        virtual ~TensorOneD() = default;
        private:
            std::vector<double> m_vec;
            std::shared_ptr<TensorMath> m_math;
    };
}
#endif /* TENSORONED_HPP_INCLUDE */