ArrayFire 的整数输入矩阵乘法

ArrayFire's matrix multiplication with integer inputs

我想通过向量乘法来执行矩阵,但是在具有整数类型的数组上以避免额外的转换(当两边都是 f32 类型时它工作正常);

这样的操作会抛出整数类型的错误类型异常,有什么解决方法吗?

ArrayFire 在现有 BLAS 实现之上实现矩阵乘法,例如 cuBLAS、clBLAS、CLBlast、FFTW3、Intel MKL 等,具体取决于后端。据我所知,BLAS API 不支持整数类型,因此任何下游项目都不支持整数类型。到目前为止,我们的项目页面上还没有任何对 matmul 整数支持的请求。欢迎您提出要求。

作为变通方法,您可以在传递给 matmul 函数之前在 arrayfire 数组上调用 array::as(f32) 方法。