pytorch 個人使用筆記

2021-10-08 00:17:42 字數 952 閱讀 7339

debug

主要記錄遇到的bug和函式的筆記,減少重複搜尋

torch.eye(n, m=none, out=none)得到單位矩陣,即對角線為1,不設定 m 預設方陣

torch.cat((a, b), dim)按維度拼接

torch.sum(input, list: dim, bool: keepdim=false, dtype=none)input:輸入乙個tensor,dim:要求和的維度,可以是乙個列表,keepdim:求和之後這個dim的元素個數為1,預設被squeeze掉,如果要保留這個維度,則應當keepdim=true

tensor.expand(n, m)只要是單維度均可進行擴充套件,但是若非單維度會報錯,單個維度的資料重複

tensor.view()把原先tensor中的資料按照行優先的順序排成乙個一維的資料(這裡應該是因為要求位址是連續儲存的),然後按照引數組合成其他維度的tensor。某些維度設為 -1 則是按需調整

報錯:runtimeerror: expected object of type torch.doubletensor but found type torch.floattensor for argument #2 『mat2』

解決:看看是不是使用了torch.tensor(),將其更改為torch.tensor(),因為torch.tensor()torch.floattensor()的別名,可以得到 float 型的,但是torch.tensor()根據原始資料型別生成相應的torch.longtensor,torch.floattensor,torch.doubletensor。

git個人使用筆記

記錄在使用git時用到的 一 遠端倉庫 第1步 建立ssh key。在使用者主目錄下,看看有沒有.ssh目錄,如果有,再看看這個目錄下有沒有id rsa和id rsa.pub這兩個檔案,如果已經有了,可直接跳到下一步。如果沒有,開啟shell windows下開啟git bash 建立ssh key...

pandas 個人使用筆記

主要是使用 pandas 過程中問題的記錄,避免重複搜尋 讀檔案 val x pd.read csv val feature.csv 分塊讀取 reader pd.read csv file,iterator true,error bad lines false chunk reader.get c...

pytorch學習使用筆記(持續更新)

model.eval 和torch.zero grad 兩者區別 在pytorch中進行validation時,會使用model.eval 切換到測試模式,在該模式下,主要用於通知dropout層和batchnorm層在train和val模式間切換 在train模式下,dropout網路層會按照設定...