shape=[2,2,3]的三维矩阵,可以视为2个shape=[2,3]的二维矩阵堆叠在一起!!最后两维才是有数据的矩阵,前面的维度只是矩阵的排列而已!
所以在矩阵运算的时候,其实最后都可以转成我们常见的二维矩阵运算,遵循的原则是:在多维矩阵相乘中,需最后两维满足shape匹配原则,最后两维才是有数据的矩阵,前面的维度只是矩阵的排列而已!
比如说是一个输入[64,3,32,32],第一维度表示batch_size,第二维度表示通道数。
a = 
[[[ 1.  2.  3.]
  [ 4.  5.  6.]]
 [[ 7.  8.  9.]
  [10. 11. 12.]]]
b = 
[[[ 1.  2.]
  [ 3.  4.]
  [ 5.  6.]]
 [[ 7.  8.]
  [ 9. 10.]
  [11. 12.]]]


 
                     
                     
                        
                        