TensorShape.hpp

Go to the documentation of this file.

#pragma once
 
#include <vector>
#include <FAST/Object.hpp>
#include <FAST/Data/DataTypes.hpp>
 
namespace fast {
 
class FAST_EXPORT TensorShape {
    public:
        TensorShape(std::initializer_list<int> dimensions);
        TensorShape(VectorXi dimensions);
        explicit TensorShape(std::vector<int> dimensions);
        TensorShape(const TensorShape& other);
        TensorShape& operator=(const TensorShape& other);
        TensorShape();
        bool empty() const;
        int getTotalSize() const;
        int getDimensions() const;
        int getKnownDimensions() const;
        int getUnknownDimensions() const;
        std::vector<int> getAll() const;
        int& operator[](int i);
        const int& operator[](int i) const;
        void setDimension(int i, int value);
        void addDimension(int value);
        void insertDimension(int position, int value);
        std::string toString() const;
 
        void deleteDimension(int index);
 
        void deleteDimensions(int startIndex, int endIndex);
    private:
        std::vector<int> m_data;
};
 
}