Python按特定轴求和

1.文字解释

按特定轴求和,即设定axis的取值。

axis=0表示将原矩阵的第0维度丢弃

axis=1表示将原矩阵的第1维度丢弃

axis=i表示将原矩阵的第i维度丢弃。

具体的可参阅下图:

Python按特定轴求和

2.python 帮助理解

import torch
a = torch.ones((2,5,4))
a

#运行结果
tensor([[[1., 1., 1., 1.],
         [1., 1., 1., 1.],
         [1., 1., 1., 1.],
         [1., 1., 1., 1.],
         [1., 1., 1., 1.]],

        [[1., 1., 1., 1.],
         [1., 1., 1., 1.],
         [1., 1., 1., 1.],
         [1., 1., 1., 1.],
         [1., 1., 1., 1.]]])

 axis=0

Python按特定轴求和

  axis=1

Python按特定轴求和

axis=[0,2]

Python按特定轴求和

 

 

注意!!

如果设定keepdim=True,则相应的axis=i的第2个维度不再是直接丢弃(消失),而是变成了1,具体可参见下图:

Python按特定轴求和

python 帮助理解

axis=0,keepdim=True

Python按特定轴求和

 

axis=1,keepdim=True

Python按特定轴求和

 

axis=[0,2],keepdim=True

不再是第0维度,第2维度丢弃,维度变成了1

Python按特定轴求和

 

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
上一篇 2022年6月1日 下午12:15
下一篇 2022年6月1日 下午12:18

相关推荐

本站注重文章个人版权,不会主动收集付费或者带有商业版权的文章,如果出现侵权情况只可能是作者后期更改了版权声明,如果出现这种情况请主动联系我们,我们看到会在第一时间删除!本站专注于人工智能高质量优质文章收集,方便各位学者快速找到学习资源,本站收集的文章都会附上文章出处,如果不愿意分享到本平台,我们会第一时间删除!