python類參數(shù)定義及數(shù)據(jù)擴(kuò)展方式unsqueeze/expand
類的參數(shù)定義
將conda環(huán)境設(shè)置為ai,conda activate ai
這個(gè)文件的由來(lái):
由于在yolov1的pytorch實(shí)現(xiàn)的損失函數(shù)中,看到繼承了nn.Module,并且其中兩個(gè)參數(shù)不像c++那里指定類型,那么他們的類型是哪里來(lái)的
這里就是在探索這樣一件事
操作邏輯:
- 先在類中定義了構(gòu)造函數(shù)以及一個(gè)自定義函數(shù);
- 構(gòu)造函數(shù)定義了屬性S、B,自定義函數(shù)引入兩個(gè)參數(shù),對(duì)兩個(gè)參數(shù)進(jìn)行調(diào)用
- 這里就說(shuō)明參數(shù)的結(jié)構(gòu)是怎么樣的,取決于參數(shù)被調(diào)用了什么東西,比如這里調(diào)用了
N = box1.size(0) M = box2.size(0)
說(shuō)明了它是類似一個(gè)矩陣的東西,對(duì)應(yīng)的box1的定義就是`torch.rand(10,4)
- 這里就說(shuō)明參數(shù)的結(jié)構(gòu)是怎么樣的,取決于參數(shù)被調(diào)用了什么東西,比如這里調(diào)用了
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)用方式就表示了變量是什么類型,這里是一個(gè)張量,其中每個(gè)元素是一個(gè)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ù)擴(kuò)展
探究unsqueeze以及expand的使用方法,unsqueeze可以增加一個(gè)緯度,但是維度的siz只是1而已,而expand就可以將數(shù)據(jù)進(jìn)行復(fù)制,將數(shù)據(jù)變?yōu)閚
# 獲得一開(kāi)始的初始化數(shù)值:tensor([[a1,a2,a3]]) nn1=torch.rand(1,3) print(nn1) # unsqueeze是解壓的意思,在第i個(gè)維度上進(jìn)行擴(kuò)展,將其擴(kuò)展為tensor([[[a1,a2,a3]]]) nn1=nn1.unsqueeze(0) print("*"*100) print(nn1) #利用expand對(duì)數(shù)據(jù)進(jìn)行擴(kuò)展 nn1=nn1.expand(1,3,3) print("*"*100) print(nn1)
到此這篇關(guān)于python類參數(shù)定義及數(shù)據(jù)擴(kuò)展方式unsqueeze/expand的文章就介紹到這了,更多相關(guān)python unsqueeze/expand內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Python學(xué)習(xí)筆記之迭代器和生成器用法實(shí)例詳解
這篇文章主要介紹了Python學(xué)習(xí)筆記之迭代器和生成器用法,結(jié)合實(shí)例形式詳細(xì)分析了Python迭代器與生成器的功能、原理、定義及使用方法,需要的朋友可以參考下2019-08-08pytorch常用函數(shù)定義及resnet模型修改實(shí)例
這篇文章主要為大家介紹了pytorch常用函數(shù)定義及resnet模型修改實(shí)例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-06-06Python中的pyecharts庫(kù)使用總結(jié)
這篇文章主要介紹了Python中的pyecharts庫(kù)使用總結(jié),Pyecharts 提供了一個(gè)簡(jiǎn)單而直觀的 API 接口,使得使用者無(wú)需了解復(fù)雜的 JavaScript 語(yǔ)法,即可通過(guò) Python 代碼實(shí)現(xiàn)高度定制化的圖表設(shè)計(jì),需要的朋友可以參考下2023-12-12Python matplotlib繪制散點(diǎn)圖的實(shí)例代碼
這篇文章主要給大家介紹了關(guān)于Python matplotlib繪制散點(diǎn)圖的相關(guān)資料,所謂散點(diǎn)圖就是反映兩組變量每個(gè)數(shù)據(jù)點(diǎn)的值,并且從散點(diǎn)圖可以看出它們之間的相關(guān)性,需要的朋友可以參考下2021-06-06python實(shí)現(xiàn)簡(jiǎn)單的五子棋游戲
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)簡(jiǎn)單的五子棋游戲,文中示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2020-09-09No module named ‘win32gui‘ 的解決方法(踩坑之旅)
這篇文章主要介紹了No module named ‘win32gui‘ 的解決方法(踩坑之旅),文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-02-02Python實(shí)現(xiàn)蒙特卡洛算法小實(shí)驗(yàn)過(guò)程詳解
這篇文章主要介紹了Python實(shí)現(xiàn)基于蒙特卡洛算法過(guò)程詳解,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-07-07Python?matplotlib實(shí)戰(zhàn)之箱型圖繪制
箱型圖(Box?Plot),也稱為盒須圖或盒式圖,是一種用作顯示一組數(shù)據(jù)分布情況的統(tǒng)計(jì)圖,因型狀如箱子而得名,本文主要為大家介紹了如何使用Matplotlib繪制箱型圖,需要的小伙伴可以參考下2023-08-08