GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-14 12:46:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

dot: handle the rowvector.dot(colvector) case where one needs to transpose.

Browse Source
This commit is contained in:
Benoit Jacob 2010-02-27 11:19:14 -05:00
parent 15a33622ac
commit 3f393490ad
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
Eigen/src/Core/Dot.h
View File

@ -29,7 +29,15 @@
// with mismatched types, the compiler emits errors about failing to instantiate cwiseProduct BEFORE // with mismatched types, the compiler emits errors about failing to instantiate cwiseProduct BEFORE
// looking at the static assertions. Thus this is a trick to get better compile errors. // looking at the static assertions. Thus this is a trick to get better compile errors.
template<typename T, typename U, template<typename T, typename U,
bool IsSameType = ei_is_same_type<typename T::Scalar, typename U::Scalar>::ret> bool IsSameType = ei_is_same_type<typename T::Scalar, typename U::Scalar>::ret,
// the NeedToTranspose condition here is taken straight from Assign.h
bool NeedToTranspose = T::IsVectorAtCompileTime
&& U::IsVectorAtCompileTime
&& ((int(T::RowsAtCompileTime) == 1 && int(U::ColsAtCompileTime) == 1)
| // FIXME | instead of || to please GCC 4.4.0 stupid warning "suggest parentheses around &&".
// revert to || as soon as not needed anymore.
(int(T::ColsAtCompileTime) == 1 && int(U::RowsAtCompileTime) == 1))
>
struct ei_dot_nocheck struct ei_dot_nocheck
{ {
static inline typename ei_traits<T>::Scalar run(const MatrixBase<T>& a, const MatrixBase<U>& b) static inline typename ei_traits<T>::Scalar run(const MatrixBase<T>& a, const MatrixBase<U>& b)
@ -39,7 +47,16 @@ struct ei_dot_nocheck
}; };
template<typename T, typename U> template<typename T, typename U>
struct ei_dot_nocheck<T, U, false> struct ei_dot_nocheck<T, U, true, true>
{
static inline typename ei_traits<T>::Scalar run(const MatrixBase<T>& a, const MatrixBase<U>& b)
{
return a.adjoint().cwiseProduct(b).sum();
}
};
template<typename T, typename U, bool NeedToTranspose>
struct ei_dot_nocheck<T, U, false, NeedToTranspose>
{ {
static inline typename ei_traits<T>::Scalar run(const MatrixBase<T>&, const MatrixBase<U>&) static inline typename ei_traits<T>::Scalar run(const MatrixBase<T>&, const MatrixBase<U>&)
{ {