首页 > 其他分享 >关于初学Cs231n

关于初学Cs231n

时间:2022-09-18 21:33:19浏览次数:84  
标签:Cs231n argmin sum 初学 关于 np array Out axis

Python函数:np.sum()以及axis=0、axis=1用法

常用于矩阵求和计算,以下用法分为三种情况来介绍!
格式:np.sum(a)
np.sum(a, axis=0) ------->列求和
np.sum(a, axis=1) ------->行求和
注意:特别注意后两种用法。

In :           a = np.array([[0, 2, 1], [3, 5, 6], [0, 1, 1]]) #得到一个矩阵a
Out:          [[0 2 1]
              [3 5 6]
              [0 1 1]]                                         
                                                             
In :           a.sum()                                         # 将a中所有元素求和 0+2+1+3+5+6+0+1+1=19
Out:         19
 
In :         np.sum(a,axis=1)                                  # 求出每行元素的和
Out:         [ 3  14  2]                                      

In :         np.sum(a,axis=0)                                  # 求出每列元素的和
Out:         [3  8  8]                                         

numpy.argmin(a, axis=None, out=None)[source]

给出axis方向最小值的下标

>>> a = np.arange(6).reshape(2,3)
>>> a
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.argmin(a)
0
>>> np.argmin(a, axis=0)
array([0, 0, 0])
>>> np.argmin(a, axis=1)
array([0, 0])

Parameters:
a : Input array.

axis : 默认将输入数组展平。否则,按照axis方向

out : 可选

Returns:
index_array : 下标组成的数组。shape与输入数组a去掉axis的维度相同。

标签:Cs231n,argmin,sum,初学,关于,np,array,Out,axis
From: https://www.cnblogs.com/bigcoding/p/16705876.html

相关文章