python類參數(shù)定義及數(shù)據(jù)擴展方式unsqueeze/expand

更新時間：2022年08月23日 10:30:20 作者：徽先生

本文主要介紹了python類參數(shù)定義及數(shù)據(jù)擴展方式unsqueeze/expand，文章通過圍繞主題展開詳細的內(nèi)容介紹，具有一定的參考價值，需要的小伙伴可以參考一下

類的參數(shù)定義

將conda環(huán)境設(shè)置為ai，conda activate ai

這個文件的由來：

由于在yolov1的pytorch實現(xiàn)的損失函數(shù)中，看到繼承了nn.Module，并且其中兩個參數(shù)不像c++那里指定類型，那么他們的類型是哪里來的

這里就是在探索這樣一件事

操作邏輯：

先在類中定義了構(gòu)造函數(shù)以及一個自定義函數(shù)；
構(gòu)造函數(shù)定義了屬性S、B，自定義函數(shù)引入兩個參數(shù)，對兩個參數(shù)進行調(diào)用
- 這里就說明參數(shù)的結(jié)構(gòu)是怎么樣的，取決于參數(shù)被調(diào)用了什么東西，比如這里調(diào)用了N = box1.size(0) M = box2.size(0)說明了它是類似一個矩陣的東西，對應的box1的定義就是`torch.rand(10,4)

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable

#探究屬性S，B是如何產(chǎn)生的，以及box1、box2是如何產(chǎn)生的、如何調(diào)用
class yoloLoss(nn.Module):
    def __init__(self,S,B):
        self.S=S
        self.B=B
    def compute_iot(self,box1,box2):
        N = box1.size(0)  #調(diào)用方式就表示了變量是什么類型，這里是一個張量，其中每個元素是一個tensor，所以是N*4的張量
        M = box2.size(0)
        print(M,N)

yoloLoss1 =yoloLoss(10, 11)
yoloLoss1.compute_iot(torch.rand(10,4),torch.rand(11,4))

數(shù)據(jù)擴展

探究unsqueeze以及expand的使用方法，unsqueeze可以增加一個緯度，但是維度的siz只是1而已，而expand就可以將數(shù)據(jù)進行復制，將數(shù)據(jù)變?yōu)閚

# 獲得一開始的初始化數(shù)值：tensor([[a1,a2,a3]])
nn1=torch.rand(1,3)
print(nn1)
# unsqueeze是解壓的意思，在第i個維度上進行擴展，將其擴展為tensor([[[a1,a2,a3]]])
nn1=nn1.unsqueeze(0)
print("*"*100)
print(nn1)
#利用expand對數(shù)據(jù)進行擴展
nn1=nn1.expand(1,3,3)
print("*"*100)
print(nn1)

到此這篇關(guān)于python類參數(shù)定義及數(shù)據(jù)擴展方式unsqueeze/expand的文章就介紹到這了,更多相關(guān)python unsqueeze/expand內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

相關(guān)文章

如何從csv文件構(gòu)建Tensorflow的數(shù)據(jù)集
這篇文章主要介紹了如何從csv文件構(gòu)建Tensorflow的數(shù)據(jù)集，幫助大家更好的理解和使用Tensorflow，感興趣的朋友可以了解下
2020-09-09
教你用Python尋找重復文件并刪除的腳本寫法
這篇文章主要介紹了如何用Python尋找重復文件并刪除,該腳本主要包括diskwalk,chechsum,find_dupes,delete模塊，其中diskwalk模塊是遍歷文件的，給定路徑，遍歷輸出該路徑下的所有文件，需要的朋友可以參考下
2022-01-01
windows 10 設(shè)定計劃任務自動執(zhí)行 python 腳本的方法
這篇文章主要介紹了windows 10 設(shè)定計劃任務自動執(zhí)行 python 腳本的方法,本文圖文并茂給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友可以參考下
2019-09-09
簡單了解Python多態(tài)與屬性運行原理
這篇文章主要介紹了簡單了解Python多態(tài)與屬性運行原理,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2020-06-06
梅爾倒譜系數(shù)（MFCC）實現(xiàn)
這篇文章主要為大家詳細介紹了梅爾倒譜系數(shù)(MFCC)實現(xiàn)，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2019-06-06
python 圖片二值化處理（處理后為純黑白的圖片）
這篇文章主要介紹了python 圖片二值化處理（處理后為純黑白的圖片），文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2019-11-11
python字典保存為json后讀取出錯問題及解決
這篇文章主要介紹了python字典保存為json后讀取出錯問題及解決,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
2024-02-02
Python自動化測試工具Splinter簡介和使用實例
這篇文章主要介紹了Python自動化測試工具Splinter簡介和使用實例,Splinter可以非常棒的模擬瀏覽器的行為，Splinter提供了豐富的API，可以獲取頁面的信息判斷當前的行為所產(chǎn)生的結(jié)果
2014-05-05
python編程普通及類和靜態(tài)方法示例詳解
普通方法會將實例傳入方法當中（通常用self表示），類方法會將類傳入方法當中（通常用cls表示），靜態(tài)方法中傳入與類無關(guān)的變量。下面將舉例詳細說明
2021-10-10
Python常用特殊方法實例總結(jié)
這篇文章主要介紹了Python常用特殊方法,結(jié)合實例形式總結(jié)分析了Python常見的__init__、__new__、__del__、__str__、__repr__等特殊方法與描述符相關(guān)功能及使用技巧,需要的朋友可以參考下
2019-03-03