官方文档地址
inf
代表无穷
import numpy as np
print(np.inf)
print(np.log(0))
argmax
numpy.argmax(a, axis=None, out=None)
返回在axis轴上的最大值的下标
- a:输入的数组
- axis:optional,如果没有提供这个值,那么函数会将数组reshape为一维然后返回这个一维数组的最大值的下标,否则返回沿指定坐标轴下的最大值
import numpy as np
a=np.arange(6).reshape(2,3)
print(a)
print(np.argmax(a))
print(np.argmax(a,axis=0))
print(np.argmax(a,axis=1))
输出:
[[0 1 2]
[3 4 5]]
5
[1 1 1]
[2 2]
dot
numpy.dot(a, b, out=None)
进行两个数组的点乘
- 当a和b均为一维数组时,是数组内积
import numpy as np
a=[1,2,3]
x=np.array(a)
y=np.array(a)
print(np.dot(x,y))
输出:
14
- 当a和b均为二维数组时,进行的是矩阵相乘
import numpy as np
a=[[1,1,1],
[1,1,1]]
b=[[1,1],
[1,1],
[1,1]]
x=np.array(a)
y=np.array(b)
print(x.shape)
print(y.shape)
print(np.dot(x,y).shape)
输出:
(2, 3)
(3, 2)
(2, 2)