代码之家 › 专栏 › 技术社区 › Hari Krishnan

python中的矩阵点积

numpy arrays python

-1

Hari Krishnan · 技术社区 · 6 年前

我是python新手,对代码感到困惑

X = np.array([2,3,4,4])
print(np.dot(X,X))

这很管用

Y = np.array([[100],
        [200],
        [300],
        [400]])
print(np.dot(Y,Y))

但事实并非如此。我理解这是因为与数组维度的关系。但我不明白怎么做。请解释一下。

1 回复 | 直到 6 年前

hpaulj 6 年前

X 是1d数组(行向量不是正确的描述符):

In [382]: X = np.array([2,3,4,4])
In [383]: X.shape
Out[383]: (4,)
In [384]: np.dot(X,X)     # docs for 1d arrays apply
Out[384]: 45

Y 是二维阵列。

In [385]: Y = X[:,None]
In [386]: Y
Out[386]: 
array([[2],
       [3],
       [4],
       [4]])
In [387]: Y.shape
Out[387]: (4, 1)
In [388]: np.dot(Y,Y)
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-388-3a0bc5156893> in <module>()
----> 1 np.dot(Y,Y)

ValueError: shapes (4,1) and (4,1) not aligned: 1 (dim 1) != 4 (dim 0)

对于二维阵列,第一对的最后一个维度是第二对到第二对的最后一个维度。

In [389]: np.dot(Y,Y.T)    # (4,1) pair with (1,4) to produce (4,4)
Out[389]: 
array([[ 4,  6,  8,  8],
       [ 6,  9, 12, 12],
       [ 8, 12, 16, 16],
       [ 8, 12, 16, 16]])
In [390]: np.dot(Y.T,Y)    # (1,4) pair with (4,1) to produce (1,1)
Out[390]: array([[45]])

推荐文章

July · 如何定义数字间隔,然后四舍五入

1 年前

Community wiki · 对象名称前的单下划线和双下划线的含义是什么?

1 年前