Generic/Tensor.hpp

Go to the documentation of this file.

 
#ifndef LLU_CONTAINERS_GENERIC_TENSOR_HPP
#define LLU_CONTAINERS_GENERIC_TENSOR_HPP
 
#include "LLU/Containers/Generic/Base.hpp"
#include "LLU/Containers/Interfaces.h"
 
namespace LLU {
 
    template<>
    class MContainer<MArgumentType::Tensor>;
 
    using GenericTensor = MContainer<MArgumentType::Tensor>;
    
    template<>
    class MContainer<MArgumentType::Tensor> : public TensorInterface, public MContainerBase<MArgumentType::Tensor> {
    public:
        using MContainerBase<MArgumentType::Tensor>::MContainerBase;
 
        MContainer() = default;
 
        MContainer(mint type, mint rank, const mint* dims);
 
        MContainer clone() const {
            return MContainer {cloneContainer(), Ownership::Library};
        }
 
        mint getRank() const override {
            return LibraryData::API()->MTensor_getRank(this->getContainer());
        }
 
        mint const* getDimensions() const override {
            return LibraryData::API()->MTensor_getDimensions(this->getContainer());
        }
 
        mint getFlattenedLength() const override {
            return LibraryData::API()->MTensor_getFlattenedLength(this->getContainer());
        }
 
        mint type() const override {
            return LibraryData::API()->MTensor_getType(this->getContainer());
        }
 
        void* rawData() const override;
 
    private:
 
        mint shareCountImpl() const noexcept override {
            return LibraryData::API()->MTensor_shareCount(this->getContainer());
        }
 
        void passImpl(MArgument& res) const noexcept override {
            MArgument_setMTensor(res, this->getContainer());
        }
 
        Container cloneImpl() const override;
    };
 
}  // namespace LLU
 
#endif    // LLU_CONTAINERS_GENERIC_TENSOR_HPP