理解Numpy中axis的正确姿势

分类: 文章 • 2024-02-07 17:29:40

写这个博客的关键起因：np.any()这个函数，其中参数axis的指定让我好一阵头脑风暴。除了在np.any()中，很多numpy的API都有axis这个变量，那么到底如何理解呢？在一番思考后，总结下来给大家分享。

当我们创建一个三维矩阵的时候，如图中的矩阵维度为(2, 3, 3)，axis所对应的维度如图所示。