python中，axis=0，axis=1，axis=2的理解【对于按待定轴求和，axis=‘x’的理解】

news2024/9/21 0:34:54

一、axis

1.假设我们有一个5行4列的矩阵，见下图所示。
这个矩阵的shape：[5][4],维度是2，axis：0,1，按照行，轴为0，按照列轴为1。
当我们按照axis=0，求sum时：我们得到的shape为[4]
当我们按照axis=1，求sum时：我们得到的shape为[5]
在这里插入图片描述

2.技巧：
axis=几，就把对应轴的维度去掉。例如：
假设我们的矩阵shape[2,5,4]
当 axis=1 求和时，得到矩阵的shape[2,4]
当 axis=0 求和时，得到矩阵的shape[5,4]
当 axis=[0,2] 求和时，得到矩阵的shape[5]

二、keepdims

1.keepdims ：在进行某些数组/矩阵操作时的一个参数，通常用于指示操作的结果是否应该保留原始数组的维度信息，默认值为false。
2.假设我们的矩阵shape[2,5,4]
当 axis=1 ,keepdims=True求和时，得到矩阵的shape[2,1,4]
当 axis=2 ,keepdims=True求和时，得到矩阵的shape[2,5,1]
当 axis=[1,2] ,keepdims=True求和时，得到矩阵的shape[2,1,1]

三、代码演示

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

运行结果：
在这里插入图片描述

1.axis

a.sum(axis=1).shape

运行结果：
在这里插入图片描述

a.sum(axis=0).shape

运行结果：
在这里插入图片描述

a.sum(axis=[0,2]).shape

运行结果：
在这里插入图片描述

2.keepdims

a.sum(axis=1,keepdims=True).shape

运行结果：
在这里插入图片描述

a.sum(axis=2,keepdims=True).shape

运行结果：
在这里插入图片描述

a.sum(axis=[1,2],keepdims=True).shape

运行结果：
在这里插入图片描述

☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺☺
我是韩一，用知识认识更多的人，欢迎大家指正！

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/1229662.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！