Numpy 中的线形代数模块

主要为将来的张量(Tensor)计算打基础

NumPy 包包含numpy.linalg模块，提供线性代数所需的所有功能,主要功能有下

dot                     两个数组的点积
vdot                    两个向量的点积
inner                   两个数组的内积
matmul                  两个数组的矩阵积

1 点积 numpy.dot()
对于一维数组(向量),相当于内积;对于二维数组(矩阵),相当于矩阵乘法;对于多维数组(张量),它是a的最后一个轴上的和与b的倒数第二个轴的乘积。

import numpy as np

a = np.array([[[1,2],[3,4]],[[5,6],[7,8]]])
b = np.array([[[2,4],[3,6]],[[4,8],[5,10]]])
c = np.dot(a,b)

print(a)
print(b)
print()
print(c)
# 输出
[[[1 2]
  [3 4]]