对多维数组计算中axis的理解

本文通过举例和树状图的方式,详细解释了在多维数组计算中,特别是使用tf.reduce_mean()时,如何理解axis参数。对于二维矩阵,axis=0表示按行求均值,axis=1表示按列求均值。将多维矩阵视为树结构,有助于理解高维情况下的axis。文章还探讨了四维矩阵的例子,并提供了代码实现。
摘要由CSDN通过智能技术生成

对多维数组计算中axis的理解

以tf.reduce_mean()为例

axis参数用来指定操作轴,对于二维矩阵来说是比较好理解的,比如上面的求均值函数,axis=0为每行的对应元素求均值,axis=1为每列的对应元素求均值。但是个人感觉这样记忆一是容易记混,二是难以拓展到高维。所以想到了一种理解方式,供大家参考。

我这种理解其实就是对于上面加粗的对应元素的扩展,二维时候是axis=0是每行的第一个元素求均值、第二个元素……,axis=1是每列的第一个元素求均值,第二个元素……

扩展到多维其实也是一样,我把多维矩阵理解成一个树状图,首先还是以简单的二维矩阵为例

二维矩阵

这样一个矩阵,其实也可以用下面的形式表示

树状表示
根节点表示该矩阵,根节点下面的一行(本文中称为第一行)表示行维度的两个元素,行维度的元素可以看做是向量,每个向量又包含两个列维度的元素。这样的话,如果令axis=0,计算每行中第一个元素的均值就是下图这样:二维均值计算

<
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值