python三维、四维数组相乘

没找到规则,只能通过具体例子理解,有知道具体规则的望不吝赐教。
例1,三维数组相乘

import numpy as np
a=np.zeros((2,2,2))
a[:,:,0]=([[3,6],[5,8]])
a[:,:,1]=([[2,5],[7,2]])
b=np.zeros((2,2,2))
b[:,:,0]=([[3,2],[9,6]])
b[:,:,1]=([[7,8],[1,0]])
c=a*b

则三维矩阵a、b的两页分别是:

a(:,:,0)=(3568),a(:,:,1)=(2752)


b(:,:,0)=(3926),b(:,:,1)=(7180)

则c=a*b为a、b的两页分别相乘
(注意:这里是点乘),得结果为


c(:,:,0)=(9451248),c(:,:,1)=(147400)

可看出c的维数为(2,2,2)。

例2,四维数组相乘

import numpy as np
a=np.zeros((2,2,2,2))
a[:,:,0,0]=([[0,8],[3,6]])
a[:,:,0,1]=([[5,1],[4,7]])
a[:,:,1,0]=([[9,2],[5,7]])
a[:,:,1,1]=([[2,0],[3,9]])
b=np.zeros((2,1,1,2))
b[:,0,0,0]=([3,2])
b[:,0,0,1]=([9,4])
c=a*b

自定义称第四维为张,则四维矩阵a、b对应的不同张、页分别是:

a(:,:,0,0)=(0386),a(:,:,0,1)=(5417)


a(:,:,1,0)=(9527),a(:,:,1,1)=(2309)


b(:,0,0,0)=(32),b(:,0,0,1)=(94)

将a的每张每页与b的每张每页对应相乘
(注意:这里仍是点乘),b在由两行一列补成了两行两列进行相乘。


c(:,:,0,0)=(062412),c(:,:,0,1)=(4516928)


c(:,:,1,0)=(2710614),c(:,:,1,1)=(1812036)

c的维度为(2,2,2,2)。

    原文作者:大整数乘法问题
    原文地址: https://blog.csdn.net/u013645510/article/details/53808514
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞