axis 0 1 2和 dim 0 1 2计算过程

在这里插入图片描述

三维度,dim=0 相加,元素如上图,【3,2,3】 变成【2,3】
三个两行三列,变成 两行三列的数据

在这里插入图片描述【3,2,3】维度
dim =0,1,2
在哪一个维度操作,维度消失
【3,2,3】

三维有点绕,

发布了98 篇原创文章 · 获赞 141 · 访问量 26万+

猜你喜欢

转载自blog.csdn.net/m0_37192554/article/details/103514431