numpy的三维数组的 “点积” 是怎么计算的？？大佬求助！！

kerwin_lui

import numpy as np

a = np.random.randint(0, 10, (2, 3, 2))
b = np.random.randint(0, 10, (2, 2, 3))
c = np.dot(a, b)
print(a, b, c)

输出：
a = [[[2 8]
  [8 4]
  [0 5]]

[[8 7]
  [3 4]
  [9 3]]]

b = [[[7 1 0]
  [0 6 8]]

[[9 9 2]
  [9 2 2]]]

c = [[[[ 14  50  64]
   [ 90  34  20]]

  [[ 56  32  32]
   [108  80  24]]

  [[  0  30  40]
   [ 45  10  10]]]


[[[ 56  50  56]
   [135  86  30]]

  [[ 21  27  32]
   [ 63  35  14]]

  [[ 63  27  24]
   [108  87  24]]]]



问题：
为什么结果是这样的？
二维的矩阵相乘我懂， 但三维的压根看不懂！！！

claws0n

c = a * b
不好意思，这个是逐点乘积

alltolove

就是a和b两个数组每个位置相乘

claws0n

好像发现问题了
(2,3,2)
(2,2,3) 不匹配呀~~

kerwin_lui

claws0n 发表于 2018-10-14 11:20
好像发现问题了
(2,3,2)
(2,2,3) 不匹配呀~~

?? 哪里不匹配了？

网上说左边数组的第2维（维度从0开始数）和右边数组的第1维对上就好了~~~

kerwin_lui

alltolove 发表于 2018-10-14 09:00
就是a和b两个数组每个位置相乘

每个位置不对啊，第二个数就没对上。。。？？？

claws0n

kerwin_lui 发表于 2018-10-16 22:59
?? 哪里不匹配了？

网上说左边数组的第2维（维度从0开始数）和右边数组的第1维对上就好了~~~

那个计算有问题

1. import numpy as np
   
2. import random
   
3. 
   
4. a = np.array([[[2,8],
   
5.                [8,4],
   
6.                [0,5]],
   
7.               [[8,7],
   
8.                [3,4],
   
9.                [9,3]]])
   
10. 
    
11. b = np.array([[[7,1,0],
    
12.                [0,6,8]],
    
13.               [[9,9,2],
    
14.                [9,2,2]]])
    
15. 
    
16. ##c = np.dot(a, b)       #错
    
17. ##c = np.inner(a,b)      #错
    
18. ##c = np.einsum("ijk,klm -> ijlm",a,b)    #对
    
19. ##c = np.tensordot(a,b,axes =1)            #对
    
20. 
    
21. c = np.zeros((2,3,2,3), dtype = int)
    
22. for i in range(2):
    
23.     for j in range(3):
    
24.         for l in range(2):
    
25.             for m in range(3):
    
26.                 for k in range(2):
    
27.                     c[i,j,l,m] += a[i,j,k] * b[k,l,m]
    
28. print(c)

复制代码

kerwin_lui

claws0n 发表于 2018-10-17 03:55
那个计算有问题

你的意思是np.dot(a, b)就不能用点积来计算的吗？（但是那个结果要怎么解释呢？）

claws0n

kerwin_lui 发表于 2018-10-17 11:02
你的意思是np.dot(a, b)就不能用点积来计算的吗？（但是那个结果要怎么解释呢？）

不支持更高维度，三维是张量
它是拿 dot(a[0],b[0]) , dot(a[0],b[1]) , dot(a[1],b[0]) , dot(a[1],b[1]) 然后拆开
自己打印出来，比较以下就看到了~