python類參數定義及數據擴展方式是什么

這篇文章主要介紹“python類參數定義及數據擴展方式是什么”，在日常操作中，相信很多人在python類參數定義及數據擴展方式是什么問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”python類參數定義及數據擴展方式是什么”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

類的參數定義

將conda環境設置為ai，conda activate ai

這個文件的由來：

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

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

操作邏輯：

• 先在類中定義了構造函數以及一個自定義函數；

• 構造函數定義了屬性S、B，自定義函數引入兩個參數，對兩個參數進行調用

• 這里就說明參數的結構是怎么樣的，取決于參數被調用了什么東西，比如這里調用了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是如何產生的，以及box1、box2是如何產生的、如何調用
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)  #調用方式就表示了變量是什么類型，這里是一個張量，其中每個元素是一個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))

數據擴展

探究unsqueeze以及expand的使用方法，unsqueeze可以增加一個緯度，但是維度的siz只是1而已，而expand就可以將數據進行復制，將數據變為n

# 獲得一開始的初始化數值：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對數據進行擴展
nn1=nn1.expand(1,3,3)
print("*"*100)
print(nn1)

到此，關于“python類參數定義及數據擴展方式是什么”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續學習更多相關知識，請繼續關注億速云網站，小編會繼續努力為大家帶來更多實用的文章！