加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 百科 > 正文

c – MKL或BLAS例程将矢量乘以标量不合适的位置

发布时间:2020-12-16 05:06:32 所属栏目:百科 来源:网络整理
导读:我在模拟软件中工作,在阵列上完成的许多操作之一是按数字缩放矢量. 我有这样的代码: //Just some initialization code,don't bother about this partint n = 10000;std::vectordouble input(n,42.0);std::vectordouble output(input.size());double alpha =
我在模拟软件中工作,在阵列上完成的许多操作之一是按数字缩放矢量.

我有这样的代码:

//Just some initialization code,don't bother about this part
int n = 10000;
std::vector<double> input(n,42.0);
std::vector<double> output(input.size());

double alpha = 69.0;

//the actual calculation:
for (size_t i = 0; i < n; ++i) {
    output[i] = input[i] * alpha;
}

我有MKL库,所以如果我的计算是“就地”完成的,可以写下面的内容:

cblas_dscal(n,alpha,&input[0],1);

但是,这会改变输入变量,这不是我想要的.

我尝试使用mkl_domatcopy(),但这个操作非常慢.

解决方法

我想出的解决方案是调用cblas_dcopy()然后调用cblas_dscal().

它不是所有世界中最好的,但它仍然比原始循环更快.

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读