使用cublasGemmEx接口在GPU中进行int8型矩阵乘法运算
在一些诸如特征比对等等的应用场景中,特征值其实就是一个个的浮点数(float)组成的矩阵,这些浮点数每一个都是4个字节(32位),然后对这两个矩阵进行乘法计算,A * B = C,得出的C矩阵就是能够代表两个特征矩阵相似性的值。毫无疑问矩阵乘法可以在CPU中进行计算,但是,在一些需要实时性的场合下,CPU的运算速度可能就不能满足要求,而通过GPU去做矩阵乘法反而再合适不过,GPU处...
原创
2019-01-20 15:09:12 ·
3723 阅读 ·
0 评论