我有一个矩阵m
,其形状为(1024, 4096)
,矢量v
的形状为(4096,)
。但只包含0和1。当我使用np.dot(m, v)
将它们相乘时,我得到了标量积,但由于我必须经常使用几个矩阵,所以这一步很慢。
因此,我将矩阵和向量同时输入bool
。现在,点积的计算非常快,但结果不是整数或浮点数,而是bool值,这可能在技术上是正确的,但不是我需要的。
我找到的一个解决方案是np.dot(m, v.astype(int))
但是我假设m
然后也被转换为int并且速度增益消失了。