tensor_shape.h

#ifndef TENSOR_SHAPE_H
#define TENSOR_SHAPE_H
 
#include <string>
#include <vector>
 
#include "nforge/core/tensor.h"
 
struct TensorLayout;
 
class Tensor::Shape {
public:
    Shape() = default;
 
    Shape(const std::vector<size_t>& dims);
 
    Shape(const std::initializer_list<size_t>& dims);
 
    Shape(const TensorLayout& layout);
 
    bool operator==(const Tensor::Shape& other) const;
 
    bool operator!=(const Tensor::Shape& other) const;
 
    size_t getNumDims() const;
 
    size_t getNumElements() const;
 
    size_t getDim(size_t idx) const;
 
    bool isScalar() const;
 
    Tensor::Shape operator[](size_t index) const;
 
    Tensor::Shape operator[](const std::vector<size_t>& position) const;
 
    Tensor::Shape removeLeadingDimension() const;
 
    Tensor::Shape getSlice(size_t start, size_t end) const;
 
    std::string toString() const;
 
    std::vector<size_t> toVector() const;
 
    std::vector<size_t> withoutTrailingOnes() const;
 
    TensorLayout toContiguousLayout() const;
 
    std::vector<size_t> getContiguousStrides() const;
 
private:
    std::vector<size_t> m_dimensions;
};
 
#endif  // TENSOR_SHAPE_H