src/Tensor/TensorGlobalFunctions.h

0001 // This file is part of Eigen, a lightweight C++ template library
0002 // for linear algebra.
0003 //
0004 // Copyright (C) 2016 Eugene Brevdo <ebrevdo@gmail.com>
0005 //
0006 // This Source Code Form is subject to the terms of the Mozilla
0007 // Public License v. 2.0. If a copy of the MPL was not distributed
0008 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
0009
0010 #ifndef EIGEN_CXX11_TENSOR_TENSOR_GLOBAL_FUNCTIONS_H
0011 #define EIGEN_CXX11_TENSOR_TENSOR_GLOBAL_FUNCTIONS_H
0012
0013 namespace Eigen {
0014
0015 /** \cpp11 \returns an expression of the coefficient-wise betainc(\a x, \a a, \a b) to the given tensors.
0016  *
0017  * This function computes the regularized incomplete beta function (integral).
0018  *
0019  */
0020 template <typename ADerived, typename BDerived, typename XDerived>
0021 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const
0022     TensorCwiseTernaryOp<internal::scalar_betainc_op<typename XDerived::Scalar>,
0023                          const ADerived, const BDerived, const XDerived>
0024     betainc(const ADerived& a, const BDerived& b, const XDerived& x) {
0025   return TensorCwiseTernaryOp<
0026       internal::scalar_betainc_op<typename XDerived::Scalar>, const ADerived,
0027       const BDerived, const XDerived>(
0028       a, b, x, internal::scalar_betainc_op<typename XDerived::Scalar>());
0029 }
0030
0031 } // end namespace Eigen
0032
0033 #endif // EIGEN_CXX11_TENSOR_TENSOR_GLOBAL_FUNCTIONS_H