pytorch數據統計

norm-p(範數) 通過設置dim,在不同維度上進行範數計算 統計屬性 min,max,mean:最小、最大、平均 prod():累乘        sum():求和 首先會將tensor展開,從而計算索引值 argmax():返回最大值的索引 argmin():返回最小值的索引 如果需要返回指定類型的索引值,需要指定維度 dim,keepdim keepdim:保證得到的結果維度個數與原先的維
相關文章
相關標籤/搜索