如图,dim=-1,-2实际上对应的就是dim=0和1,实际上就是用数组下标的负数来表示在哪个维度上作变换。
对于图中的2维数组运用F.normalize方法,-2或-1与0或1等效,即在列或行上求2范数。
F.normalize中dim=-1的作用
于 2023-09-04 17:17:45 首次发布
如图,dim=-1,-2实际上对应的就是dim=0和1,实际上就是用数组下标的负数来表示在哪个维度上作变换。
对于图中的2维数组运用F.normalize方法,-2或-1与0或1等效,即在列或行上求2范数。