深度學習(十)——Pytorch統計屬性

norm 範數 norm(1):1範數,絕對值之和 norm(2):2範數,平方和在根號 矩陣範數:絕對值的和,再求max norm(1,dim=1):對維度1做1範數 prod:累乘 argmax,argmin:返回索引 mean,max等首先會把矩陣變爲dim=1 在每個dim=1上找最大值 keepdim表示保持原來的dim top-k——取大小爲 前k個和索引 kthvalue——第k個的
相關文章
相關標籤/搜索