如何使用 Eigen 库计算 uint8_t 数组的点积?

问题描述

我想在 C/C++ 中计算两个 uint8 类型数组的点积,简单实现为:

KafkaProducer

由于 Eigen 库被声明为使用 SIMD 以提高速度,因此我想传递两个原始 C 数组,然后分配给 Eigen 的类型,然后计算点积结果。实现如下:

uint64_t dotproduct_u8_naive(uint8_t* a,uint8_t* b,uint32_t len)
{
    uint64_t res = 0;
    for(uint32_t i=0; i<len; i++) {
        res += a[i] * b[i];
    }
    return res;
}

数组 uint64_t dotproduct_u8_eigen(uint8_t* a,uint32_t len) { Eigen::Map<Eigen::Matrix<uint8_t,1,Eigen::Dynamic,Eigen::RowMajor>> va(a,len); Eigen::Map<Eigen::Matrix<uint8_t,Eigen::RowMajor>> vb(b,len); uint64_t res = va.dot(vb); return res; } a 的元素不为零,但是 b 总是返回零,这与 dotproduct_u8_eigen 的预期结果不同。这对我来说似乎很奇怪,因为我以相同的方式实现了 float32 类型的向量点积并得到了正确的结果:

dotproduct_u8_naive()

我的问题是:float dotproduct_f32_eigen(float* a,float* b,uint32_t len) { Eigen::Map<Eigen::Matrix<float,len); Eigen::Map<Eigen::Matrix<float,len); float res = va.dot(vb); return res; } 错了吗?如何修改 dotproduct_u8_eigen() 以获得正确的结果?


我可以通过将 u8 矩阵转换为 uint64 矩阵来得到正确的结果,如下所示,这比简单的实现要慢得多:

dotproduct_u8_eigen()

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...