orthogonalize.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef pRC_TENSOR_TRAIN_ALGORITHMS_ORTHOGONALIZE_H
#define pRC_TENSOR_TRAIN_ALGORITHMS_ORTHOGONALIZE_H
 
#include <prc/core/basic/position.hpp>
#include <prc/core/tensor/declarations.hpp>
 
namespace pRC
{
    template<Position P, class X, IsTensorish R = RemoveReference<X>>
        requires(P == Position::Left)
    static inline constexpr auto orthogonalize(X &&a)
    {
        return expand(makeSeries<Index, R::Dimension - 1>(),
            [&](auto const... seq)
            {
                auto const [q, r] = qr(folding<P>(forward<X>(a)));
 
                return Tuple(eval(reshape<R::size(seq)...,
                                 RemoveReference<decltype(q)>::size(1)>(q)),
                    r);
            });
    }
 
    template<Position P, class X, IsTensorish R = RemoveReference<X>>
        requires(P == Position::Right)
    static inline constexpr auto orthogonalize(X &&a)
    {
        return expand(makeRange<Index, 1, R::Dimension>(),
            [&](auto const... seq)
            {
                auto const [l, q] = lq(folding<P>(forward<X>(a)));
 
                return Tuple(l,
                    eval(reshape<RemoveReference<decltype(q)>::size(0),
                        R::size(seq)...>(q)));
            });
    }
}
#endif // pRC_TENSOR_TRAIN_ALGORITHMS_ORTHOGONALIZE_H