Python中torch.norm()用法解析
1. 介紹
torch.norm()是對輸入的tensor求對應(yīng)的范數(shù)。tensor的范數(shù)有以下三種:
1.1 p-范數(shù)

1.2 Frobenius 范數(shù)
即,矩陣各項(xiàng)元素的絕對值平方的總和。

1.3 核范數(shù)
也即,求矩陣奇異值的和。該范數(shù)常被用于約束矩陣的低秩,對于稀疏性質(zhì)的數(shù)據(jù)而言,其矩陣是低秩且會包含大量冗余信息,這些信息可被用于恢復(fù)數(shù)據(jù)和提取特征。

2. API
def norm(input, p="fro", dim=None, keepdim=False, out=None, dtype=None):
參數(shù)解釋:
- input:輸入tensor類型的數(shù)據(jù)
- p:指定的范數(shù)
- 默認(rèn)為p=‘fro’,計(jì)算矩陣的Frobenius norm (Frobenius 范數(shù)),就是矩陣各項(xiàng)元素的絕對值平方的總和。
- p='nuc’時,是求核范數(shù),核范數(shù)是矩陣奇異值的和。(不常用)
- p為int的形式,是求p-范數(shù)。(常用)
- dim:指定在哪個維度進(jìn)行,如果不指定,則是在所有維度進(jìn)行計(jì)算
- keepdim:True or False,如果True,則保留dim指定的維度,F(xiàn)alse則不保留
- out:輸出的 tensor
- dtype:指定輸出的tensor的數(shù)據(jù)類型
3. 示例
import torch a = torch.ones(5, 2, 2) a_norm = a.norm(1, 1) print(a) print(a_norm)
輸出:

說明:
- 指定p=1,dim=1。也就是在a的第2個維度(dim=1)進(jìn)行1(p=1)范數(shù)計(jì)算。
可以發(fā)現(xiàn):a_norm的shape為(5,2),keepdim默認(rèn)設(shè)置的是False,所以第2個維度對應(yīng)的2消失了,如果keepdim=True,將會保留第3個維度,但是會變成(5,2,1)。 - 每一個元素都是為2,也就是1范數(shù)的結(jié)果。
到此這篇關(guān)于Python中torch.norm()用法解析的文章就介紹到這了,更多相關(guān)Python torch.norm()內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Django使用視圖動態(tài)輸出CSV以及PDF的操作詳解
這篇文章主要介紹了Django 如何使用視圖動態(tài)輸出 CSV 以及 PDF,我們需要用到 python 的 csv 和 reportLab 庫,通過django視圖來定義輸出我們需要的 csv 或者 pdf 文件,需要的朋友可以參考下2024-06-06
使用Python和Tesseract實(shí)現(xiàn)驗(yàn)證碼識別功能
驗(yàn)證碼識別是一個常見且實(shí)用的技術(shù)需求,尤其是在自動化測試和數(shù)據(jù)采集場景中,通過開源 OCR工具 Tesseract,結(jié)合 Python 的強(qiáng)大生態(tài),我們可以高效實(shí)現(xiàn)驗(yàn)證碼識別任務(wù),本篇博客將以詳細(xì)步驟和代碼示例,介紹如何使用 Python 和 Tesseract 實(shí)現(xiàn)驗(yàn)證碼識別2025-01-01
python鏈表的基礎(chǔ)概念和基礎(chǔ)用法詳解
這篇文章主要為大家詳細(xì)介紹了python鏈表的基礎(chǔ)概念和基礎(chǔ)用法,文中示例代碼介紹的非常詳細(xì),具有一定的參考價值,感興趣的小伙伴們可以參考一下2022-05-05
yolov5 win10 CPU與GPU環(huán)境搭建過程
這篇文章主要介紹了yolov5 win10 CPU與GPU環(huán)境搭建過程,本文給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2021-04-04
Python3中常用的處理時間和實(shí)現(xiàn)定時任務(wù)的方法的介紹
這篇文章主要介紹了Python3中常用的處理時間和實(shí)現(xiàn)定時任務(wù)的方法,包括循環(huán)執(zhí)行某個任務(wù)這樣的實(shí)現(xiàn),需要的朋友可以參考下2015-04-04
Python中join()函數(shù)多種操作代碼實(shí)例
這篇文章主要介紹了Python中join()函數(shù)多種操作代碼實(shí)例,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-01-01
基于Python Pygame實(shí)現(xiàn)的畫餅圖游戲
這篇文章主要介紹了基于Pygame實(shí)現(xiàn)一個畫餅圖游戲,可以根據(jù)鍵盤上輸入不同的數(shù)字,將圓分割成不同的幾個部分,每部分用不同的顏色來實(shí)現(xiàn)。需要的朋友可以參考一下2021-12-12

