backwards.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef pRC_TENSOR_TRAIN_TENSOR_FUNCTIONS_BACKWARDS_H
#define pRC_TENSOR_TRAIN_TENSOR_FUNCTIONS_BACKWARDS_H
 
#include <prc/core/functors/view.hpp>
#include <prc/tensor_train/tensor/views/backwards.hpp>
 
namespace pRC
{
    template<class X, TensorTrain::IsTensorish R = RemoveReference<X>>
    static inline constexpr auto backwards(X &&a)
    {
        if constexpr(IsInvocable<View, X>)
        {
            using V = RemoveReference<decltype(view(forward<X>(a)))>;
            using N = decltype(reverse(typename R::N()));
            using Ranks = decltype(reverse(typename R::Ranks()));
 
            return TensorTrain::TensorViews::Backwards<typename R::Type, N,
                Ranks, V>(view(forward<X>(a)));
        }
        else
        {
            return eval(backwards(a));
        }
    }
}
#endif // pRC_TENSOR_TRAIN_TENSOR_FUNCTIONS_BACKWARDS_H