/usr/include/boost/numeric/ublas/tensor
Edit: /usr/include/boost/numeric/ublas/tensor/operators_arithmetic.hpp (10106B)
//
// Copyright (c) 2018-2019, Cem Bassoy, cem.bassoy@gmail.com
//
// Distributed under the Boost Software License, Version 1.0. (See
// accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
//
// The authors gratefully acknowledge the support of
// Fraunhofer IOSB, Ettlingen, Germany
//
#ifndef BOOST_UBLAS_TENSOR_OPERATORS_ARITHMETIC_HPP
#define BOOST_UBLAS_TENSOR_OPERATORS_ARITHMETIC_HPP
#include "expression.hpp"
#include "expression_evaluation.hpp"
#include "multi_index_utility.hpp"
#include "functions.hpp"
#include
#include
#include
namespace boost{
namespace numeric{
namespace ublas {
template
class tensor;
template
class matrix_expression;
template
class vector_expression;
}
}
}
#define FIRST_ORDER_OPERATOR_RIGHT(OP, EXPR_TYPE_L, EXPR_TYPE_R) \
template \
auto operator OP ( boost::numeric::ublas:: EXPR_TYPE_L const& lhs, boost::numeric::ublas:: EXPR_TYPE_R const& rhs) { \
return boost::numeric::ublas::detail::make_binary_tensor_expression (lhs(), rhs(), \
[](auto const& l, auto const& r){ return l OP r; }); \
} \
FIRST_ORDER_OPERATOR_RIGHT (*, detail:: tensor_expression , vector_expression)
FIRST_ORDER_OPERATOR_RIGHT (+, detail:: tensor_expression , vector_expression)
FIRST_ORDER_OPERATOR_RIGHT (-, detail:: tensor_expression , vector_expression)
FIRST_ORDER_OPERATOR_RIGHT (/, detail:: tensor_expression , vector_expression)
FIRST_ORDER_OPERATOR_RIGHT (*, detail:: tensor_expression , matrix_expression)
FIRST_ORDER_OPERATOR_RIGHT (+, detail:: tensor_expression , matrix_expression)
FIRST_ORDER_OPERATOR_RIGHT (-, detail:: tensor_expression , matrix_expression)
FIRST_ORDER_OPERATOR_RIGHT (/, detail:: tensor_expression , matrix_expression)
#define FIRST_ORDER_OPERATOR_LEFT(OP, EXPR_TYPE_L, EXPR_TYPE_R) \
template \
auto operator OP ( boost::numeric::ublas:: EXPR_TYPE_L const& lhs, boost::numeric::ublas:: EXPR_TYPE_R const& rhs) { \
return boost::numeric::ublas::detail::make_binary_tensor_expression (lhs(), rhs(), \
[](auto const& l, auto const& r){ return l OP r; }); \
} \
FIRST_ORDER_OPERATOR_LEFT (*, vector_expression, detail:: tensor_expression)
FIRST_ORDER_OPERATOR_LEFT (+, vector_expression, detail:: tensor_expression)
FIRST_ORDER_OPERATOR_LEFT (-, vector_expression, detail:: tensor_expression)
FIRST_ORDER_OPERATOR_LEFT (/, vector_expression, detail:: tensor_expression)
FIRST_ORDER_OPERATOR_LEFT (*, matrix_expression, detail:: tensor_expression)
FIRST_ORDER_OPERATOR_LEFT (+, matrix_expression, detail:: tensor_expression)
FIRST_ORDER_OPERATOR_LEFT (-, matrix_expression, detail:: tensor_expression)
FIRST_ORDER_OPERATOR_LEFT (/, matrix_expression, detail:: tensor_expression)
template
auto operator+( boost::numeric::ublas::detail::tensor_expression const& lhs, boost::numeric::ublas::detail::tensor_expression const& rhs) {
return boost::numeric::ublas::detail::make_binary_tensor_expression (lhs(), rhs(), [](auto const& l, auto const& r){ return l + r; });
}
template
auto operator-( boost::numeric::ublas::detail::tensor_expression const& lhs, boost::numeric::ublas::detail::tensor_expression const& rhs) {
return boost::numeric::ublas::detail::make_binary_tensor_expression (lhs(), rhs(), [](auto const& l, auto const& r){ return l - r; });
// return boost::numeric::ublas::detail::make_lambda([&lhs,&rhs](std::size_t i){ return lhs(i) - rhs(i);});
}
template
auto operator*( boost::numeric::ublas::detail::tensor_expression const& lhs, boost::numeric::ublas::detail::tensor_expression const& rhs) {
return boost::numeric::ublas::detail::make_binary_tensor_expression (lhs(), rhs(), [](auto const& l, auto const& r){ return l * r; });
}
template
auto operator/( boost::numeric::ublas::detail::tensor_expression const& lhs, boost::numeric::ublas::detail::tensor_expression const& rhs) {
return boost::numeric::ublas::detail::make_binary_tensor_expression (lhs(), rhs(), [](auto const& l, auto const& r){ return l / r; });
}
// Overloaded Arithmetic Operators with Scalars
template
auto operator+(typename T::const_reference lhs, boost::numeric::ublas::detail::tensor_expression const& rhs) {
return boost::numeric::ublas::detail::make_unary_tensor_expression (rhs(), [lhs](auto const& r){ return lhs + r; });
//return boost::numeric::ublas::detail::make_lambda( [&lhs,&rhs](std::size_t i) {return lhs + rhs(i); } );
}
template
auto operator-(typename T::const_reference lhs, boost::numeric::ublas::detail::tensor_expression const& rhs) {
return boost::numeric::ublas::detail::make_unary_tensor_expression (rhs(), [lhs](auto const& r){ return lhs - r; });
}
template
auto operator*(typename T::const_reference lhs, boost::numeric::ublas::detail::tensor_expression const& rhs) {
return boost::numeric::ublas::detail::make_unary_tensor_expression (rhs(), [lhs](auto const& r){ return lhs * r; });
}
template
auto operator/(typename T::const_reference lhs, boost::numeric::ublas::detail::tensor_expression const& rhs) {
return boost::numeric::ublas::detail::make_unary_tensor_expression (rhs(), [lhs](auto const& r){ return lhs / r; });
}
template
auto operator+(boost::numeric::ublas::detail::tensor_expression const& lhs, typename T::const_reference rhs) {
return boost::numeric::ublas::detail::make_unary_tensor_expression (lhs(), [rhs] (auto const& l) { return l + rhs; } );
}
template
auto operator-(boost::numeric::ublas::detail::tensor_expression const& lhs, typename T::const_reference rhs) {
return boost::numeric::ublas::detail::make_unary_tensor_expression (lhs(), [rhs] (auto const& l) { return l - rhs; } );
}
template
auto operator*(boost::numeric::ublas::detail::tensor_expression const& lhs, typename T::const_reference rhs) {
return boost::numeric::ublas::detail::make_unary_tensor_expression (lhs(), [rhs] (auto const& l) { return l * rhs; } );
}
template
auto operator/(boost::numeric::ublas::detail::tensor_expression const& lhs, typename T::const_reference rhs) {
return boost::numeric::ublas::detail::make_unary_tensor_expression (lhs(), [rhs] (auto const& l) { return l / rhs; } );
}
template
auto& operator += (T& lhs, const boost::numeric::ublas::detail::tensor_expression &expr) {
boost::numeric::ublas::detail::eval(lhs, expr(), [](auto& l, auto const& r) { l+=r; } );
return lhs;
}
template
auto& operator -= (T& lhs, const boost::numeric::ublas::detail::tensor_expression &expr) {
boost::numeric::ublas::detail::eval(lhs, expr(), [](auto& l, auto const& r) { l-=r; } );
return lhs;
}
template
auto& operator *= (T& lhs, const boost::numeric::ublas::detail::tensor_expression &expr) {
boost::numeric::ublas::detail::eval(lhs, expr(), [](auto& l, auto const& r) { l*=r; } );
return lhs;
}
template
auto& operator /= (T& lhs, const boost::numeric::ublas::detail::tensor_expression &expr) {
boost::numeric::ublas::detail::eval(lhs, expr(), [](auto& l, auto const& r) { l/=r; } );
return lhs;
}
template
auto& operator += (boost::numeric::ublas::tensor& lhs, typename boost::numeric::ublas::tensor::const_reference r) {
boost::numeric::ublas::detail::eval(lhs, [r](auto& l) { l+=r; } );
return lhs;
}
template
auto& operator -= (boost::numeric::ublas::tensor& lhs, typename boost::numeric::ublas::tensor::const_reference r) {
boost::numeric::ublas::detail::eval(lhs, [r](auto& l) { l-=r; } );
return lhs;
}
template
auto& operator *= (boost::numeric::ublas::tensor& lhs, typename boost::numeric::ublas::tensor::const_reference r) {
boost::numeric::ublas::detail::eval(lhs, [r](auto& l) { l*=r; } );
return lhs;
}
template
auto& operator /= (boost::numeric::ublas::tensor& lhs, typename boost::numeric::ublas::tensor::const_reference r) {
boost::numeric::ublas::detail::eval(lhs, [r](auto& l) { l/=r; } );
return lhs;
}
template
auto const& operator +(const boost::numeric::ublas::detail::tensor_expression& lhs) {
return lhs;
}
template
auto operator -(boost::numeric::ublas::detail::tensor_expression const& lhs) {
return boost::numeric::ublas::detail::make_unary_tensor_expression (lhs(), [] (auto const& l) { return -l; } );
}
/** @brief Performs a tensor contraction, not an elementwise multiplication
*
*/
template
auto operator*(
std::pair< tensor_type_left const&, tuple_type_left > lhs,
std::pair< tensor_type_right const&, tuple_type_right > rhs)
{
using namespace boost::numeric::ublas;
auto const& tensor_left = lhs.first;
auto const& tensor_right = rhs.first;
auto multi_index_left = lhs.second;
auto multi_index_right = rhs.second;
static constexpr auto num_equal_ind = number_equal_indexes::value;
if constexpr ( num_equal_ind == 0 ){
return tensor_left * tensor_right;
}
else if constexpr ( num_equal_ind==std::tuple_size::value && std::is_same::value ){
return boost::numeric::ublas::inner_prod( tensor_left, tensor_right );
}
else {
auto array_index_pairs = index_position_pairs(multi_index_left,multi_index_right);
auto index_pairs = array_to_vector( array_index_pairs );
return boost::numeric::ublas::prod( tensor_left, tensor_right, index_pairs.first, index_pairs.second );
}
}
#endif