utility.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef cMHN_TT_UTILITY_H
#define cMHN_TT_UTILITY_H
 
#include <prc.hpp>
 
namespace cMHN::TT
{
    template<pRC::Size D, pRC::Size N = 2>
    constexpr auto getModeSizes()
    {
        return pRC::makeConstantSequence<pRC::Size, D, N>();
    }
 
    template<pRC::Size D, pRC::Size R, pRC::Size N = 2>
    constexpr auto getRanks()
    {
        return typename decltype(round(pRC::TensorTrain::Tensor<pRC::Float<>,
            decltype(getModeSizes<D, N>()),
            decltype(pRC::makeConstantSequence<pRC::Size, D - 1, R>())>()))::
            Ranks();
    }
}
 
#endif // cMHN_TT_UTILITY_H