apply.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef pRC_TENSOR_TRAIN_OPERATOR_FUNCTIONS_APPLY_H
#define pRC_TENSOR_TRAIN_OPERATOR_FUNCTIONS_APPLY_H
 
#include <prc/core/functors/apply.hpp>
#include <prc/core/functors/mul.hpp>
#include <prc/core/functors/restrict.hpp>
#include <prc/core/functors/transform.hpp>
#include <prc/core/tensor/operator/hint.hpp>
#include <prc/tensor_train/operator/declarations.hpp>
#include <prc/tensor_train/tensor/declarations.hpp>
 
namespace pRC
{
    template<Operator::Transform OT = Operator::Transform::None,
        Operator::Restrict OR = Operator::Restrict::None,
        Operator::Hint OH = Operator::Hint::None, class XA, class XB,
        TensorTrain::IsOperatorish RA = RemoveReference<XA>,
        class RB = RemoveReference<XB>>
        requires(TensorTrain::IsOperatorish<RB> ||
                    TensorTrain::IsTensorish<RemoveReference<XB>>) &&
        IsInvocable<Transform<OT>, XA> &&
        IsInvocable<Mul, ResultOf<Restrict<OR>, ResultOf<Transform<OT>, XA>>,
            XB>
    static inline constexpr auto apply(XA &&a, XB &&b)
    {
        return restrict<OR>(transform<OT>(forward<XA>(a))) * forward<XB>(b);
    }
}
#endif // pRC_TENSOR_TRAIN_OPERATOR_FUNCTIONS_APPLY_H