Tensor2_transform.hpp

Go to the documentation of this file.

/* Applies an arbitrary function to the Tensor2_Expr. */
 
#pragma once
 
namespace FTensor
{
  template <class A, class B, class T, int Dim0, int Dim1, char i, char j>
  class transform_Tensor2
  {
    const Tensor2_Expr<A, T, Dim0, Dim1, i, j> iterA;
    B function;
 
  public:
    T operator()(const int N1, const int N2) const
    {
      return function(iterA(N1, N2));
    }
 
    transform_Tensor2(const Tensor2_Expr<A, T, Dim0, Dim1, i, j> &a, B func)
        : iterA(a), function(func)
    {}
  };
 
  template <class A, class B, class T, int Dim0, int Dim1, char i, char j>
  Tensor2_Expr<transform_Tensor2<A, B, T, Dim0, Dim1, i, j>, T, Dim0, Dim1, i,
               j>
  transform(const Tensor2_Expr<A, T, Dim0, Dim1, i, j> &a, B function)
  {
    using TensorExpr = transform_Tensor2<A, B, T, Dim0, Dim1, i, j>;
    return Tensor2_Expr<TensorExpr, T, Dim0, Dim1, i, j>(
      TensorExpr(a, function));
  }
}