下面给出两个样例样例1：输出结果样例2:输出结果：样例1：输出结果样例2：博主最近阅读YOLO底层代码，Torch中对多数组矩阵有很多高维操作，看过一边之后，记录一下，以防忘记。这里一般分不清dim的意思举个例子：squeeze= 不会改变 squeeze= 会改变这里0, 1 ，2就分别代表 A， 1, B. 功能： 在某一位置插入一维这里面也是dim这个参数比较难理解dim的取值是[- input.dim()-1, imput.dim()]给定一个dim. input的维度input_dim为4, dim的取值[-5, 4]看一个简单用例，size表示维度大小，10是取值范围，a=[：，：，：，4]表示取a最后一维的第四个元素，即取[0,0,3],[5,6,1],[0,6,8],[…], 判断大于5为true，否则为false。
暂无评论