尝试将向量分量与数组相乘(逐元素乘法或广播)时出现错误。文档显示*的这种重载大小写应该很好:
AFAPI数组运算符*(const float&lhs,const array&rhs)将
两个数组或一个数组和一个值相乘。(const array&,const array&)
但是,根据下面的错误消息,也许vect(0)
需要进一步展平或缩小以使尺寸一致?
错误语句很清楚:
参数1的尺寸无效预期:ldims == rides
#include <arrayfire.h>
int main(int argc, char *argv[])
{
int device = argc > 1 ? atoi(argv[1]) : 0;
af::setDevice(device);
af::info();
int n = 3;
int N = 5;
// Create the arrays:
af::array matrix = af::constant(0,n,n,f32); // 3 x 3 float array of zeros
af::array vect = af::seq(1,N); // A col vector of floats: {1.0, ... ,5.0}
// Show the arrays:
af_print(matrix);
af_print(vect);
// Print a single component of the vector:
af_print(vect(0));
// This line produces the error (see below):
af_print(vect(0) * matrix); // Why doesn't this work?
// But somthing like this is fine:
af_print(1.0 * matrix);
return 0;
}
产生输出:
ArrayFire v3.3.2
ATI Radeon HD 6750M矩阵
[3 3 1 1]
0.0000 0.0000 0.0000
0.0000 0.0000 0.0000
0.0000 0.0000 0.0000 0.0000vect
[5 1 1 1]
1.0000
2.0000
3.0000
4.0000
5.0000vect(0)
[1 1 1 1]
1.0000
矩阵的af_print()的dims()输出= [3 3 1 1],而vect(0)= [1 1 1 1],这使我感到怀疑,但是我不确定如何进一步展平。有人会认为此示例是使用ArrayFire API的一种常见方式。
引发的错误异常是:
libc ++ abi.dylib:以af :: exception类型的未捕获异常终止:ArrayFire异常(无效输入大小:203):在函数getOutDims中在文件src / backend / ArrayInfo.cpp:173中
,参数1的尺寸无效预期:ldims ==游骑
在函数af :: array af :: operator *(const af :: array&,const af :: array&)
添加一个用例来说明:实际上,我是通过求和coeff(k)*(3-d数组Z的2d切片)来构造最终数组:
for (int j = 0; j<indx.dims(0); ++j)
final += coeff(indx(j)) * Z(af::span,af::span,indx(j));
我将研究使用a,gfor
但最初只是想获得正确的数值输出。还应注意向量:index
是预定义的,例如,sayindex = {1, 2, 4, 7, ...}
和元素不一定按顺序排列;这样可以选择特定的术语。
自从上次回答以来,OP就添加了一个用例,因此这就是在arrayfire中编写完全矢量化版本的方式。
array coeffs = moddims(coeff(indx), 1, 1, coeff.elements());
array final = sum(Z(span, span, indx) * tile(coeffs, Z.dims(0), Z.dims(1)), 2);
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句