在学习 numpy,torch 时,在很多运算中,我们都会遇到 axis or dim 这个参数 之前学习的时候,一个API 也就过去了,但有时候在一个具体的场景里,又会对这个 axis (dim) 需要进行 考虑,而不是直接调用 API扔进去就行,因此我们需要去搞明白这个 axis,dim,本文理解起来,非常友好,同时在最后也可以引起大家一些思考。 ok,让我们开始,首先直接给出观点 axis 或者 dim 是方向 ! 三维 import numpy as np # 生成数据 模仿图片 shape c w h i…