美文网首页
理解axis参数和dot函数

理解axis参数和dot函数

作者: Ra1nyHouse | 来源:发表于2017-03-31 21:42 被阅读75次

理解axis看下面的例子

多维数组的轴(axis=)是和该数组的size(或者shape)的元素是相对应的;

>>> np.random.seed(123)
>>> X = np.random.randint(0, 5, [3, 2, 2])
>>> print(X)

[[[5 2]
  [4 2]]

 [[1 3]
  [2 3]]

 [[1 1]
  [0 1]]]

>>> X.sum(axis=0)
array([[7, 6],
       [6, 6]])

>>> X.sum(axis=1)
array([[9, 4],
       [3, 6],
       [1, 2]])

>>> X.sum(axis=2)
array([[7, 6],
       [4, 5],
       [2, 1]])

对于dot,官方解释如下:

For 2-D arrays it is equivalent to matrix multiplication, and for 1-D arrays to inner product of vectors (without complex conjugation). For N dimensions it is a sum product over the last axis of a and the second-to-last of b:

2维是矩阵乘法,1维度向量内积,多维是第一个向量最后一维和第二个向量倒数第二维的乘积和

>>> a = np.arange(3*4*5*6).reshape((3,4,5,6))
>>> b = np.arange(3*4*5*6)[::-1].reshape((5,4,6,3))
>>> np.dot(a, b)[2,3,2,1,2,2]
499128
>>> sum(a[2,3,2,:] * b[1,2,:,2])
499128

相关文章

  • 理解axis参数和dot函数

    理解axis看下面的例子 多维数组的轴(axis=)是和该数组的size(或者shape)的元素是相对应的; 对于...

  • numpy的axis

    这是别人的理解 原文Numpy中有许多函数都带有一个参数:axis(对应于pytorch中的dim参数),用来指定...

  • numpy.cumsum()用法

    numpy.cumsum 函数参数有numpy.cumsum(a, axis=None, dtype=None, ...

  • 原创|Python|cumsum()

    一、参数numpy.cumsum(a, axis=None, dtype=None, out=None)函数的功能...

  • python释疑

    numpy模块 sum函数的参数(axis)axis 代表矩阵轴数。对于一维矩阵,0:行方向。对于二维矩阵,0:列...

  • Numpy中stack()函数的理解

    np.stack(array,axis,out=None),函数原型。其中最重要是的这个axis怎么理解的。举例说...

  • tf.argmax()函数中axis参数理解

    tf.argmax是tensorflow用numpy的np.argmax实现的,它能给出某个tensor对象在某一...

  • [Py010] 理解axis

    pandas很多函数里面都有axis参数,大部分默认为axis=0 经常被搞得晕头转向…… 瞅了一个小时,大概理清...

  • 11 Pandas的axis参数怎么理解

    11 Pandas的axis参数怎么理解? axis=0或者"index":如果是单行操作,就指的是某一行如果是聚...

  • python pandas-->删除DataFrame某行或某列

    drop()删除行列的操作,drop函数的axis参数0和1分别代表列和行。 1.删除DataFrame的’C’行...

网友评论

      本文标题:理解axis参数和dot函数

      本文链接:https://www.haomeiwen.com/subject/qnxbottx.html