transform.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef pRC_TENSOR_TRAIN_OPERATOR_FUNCTIONS_TRANSFORM_H
#define pRC_TENSOR_TRAIN_OPERATOR_FUNCTIONS_TRANSFORM_H
 
#include <prc/core/functors/view.hpp>
#include <prc/core/tensor/operator/transform.hpp>
#include <prc/tensor_train/operator/declarations.hpp>
 
namespace pRC
{
    template<Operator::Transform OT = Operator::Transform::None, class X>
        requires TensorTrain::IsOperatorish<RemoveReference<X>> &&
        (OT == Operator::Transform::Transpose)
    static inline constexpr auto transform(X &&a)
    {
        return transpose(forward<X>(a));
    }
 
    template<Operator::Transform OT = Operator::Transform::None, class X>
        requires TensorTrain::IsOperatorish<RemoveReference<X>> &&
        (OT == Operator::Transform::Adjoint)
    static inline constexpr auto transform(X &&a)
    {
        return adjoint(forward<X>(a));
    }
 
    template<Operator::Transform OT = Operator::Transform::None, class X>
        requires TensorTrain::IsOperatorish<RemoveReference<X>> &&
        (OT == Operator::Transform::None)
    static inline constexpr auto transform(X &&a)
    {
        if constexpr(!IsInvocable<View, X>)
        {
            return eval(forward<X>(a));
        }
        else
        {
            return view(forward<X>(a));
        }
    }
}
#endif // pRC_TENSOR_TRAIN_OPERATOR_FUNCTIONS_TRANSFORM_H