np.dot类型为bool但int结果?

时间:2017-04-02 09:35:22

标签: python numpy

我有一个矩阵m,其形状为(1024, 4096),矢量v的形状为(4096,)。但只包含0和1。当我使用np.dot(m, v)将它们相乘时,我得到了标量积,但由于我必须经常使用几个矩阵,所以这一步很慢。

因此,我将矩阵和向量同时输入bool。现在,点积的计算非常快,但结果不是整数或浮点数,而是bool值,这可能在技术上是正确的,但不是我需要的。

我找到的一个解决方案是np.dot(m, v.astype(int))但是我假设m然后也被转换为int并且速度增益消失了。

0 个答案:

没有答案