pytorch和numpy默認(rèn)浮點(diǎn)類型位數(shù)詳解
pytorch和numpy默認(rèn)浮點(diǎn)類型位數(shù)
numpy中默認(rèn)浮點(diǎn)類型為64位,pytorch中默認(rèn)浮點(diǎn)類型位32位
測(cè)試代碼如下
- numpy版本:1.19.2
- pytorch版本:1.2.0
In [1]: import torch In [2]: import numpy as np # 版本信息 In [3]: "pytorch version: {}, numpy version: {}".format(torch.__version__, np.__version__) Out[3]: 'pytorch version: 1.2.0, numpy version: 1.19.2' # numpy In [4]: dat_np = np.array([1,2,3], dtype="float") In [5]: dat_np.dtype Out[5]: dtype('float64') # pytorch In [6]: dat_torch = torch.tensor([1,2,3]) In [7]: dat_torch = dat_torch.float() In [8]: dat_torch.dtype Out[8]: torch.float32
pytorch和numpy的默認(rèn)類型與轉(zhuǎn)換問題
pytorch對(duì)于浮點(diǎn)類型默認(rèn)為float32,而numpy的默認(rèn)類型是float64,轉(zhuǎn)換的代碼:
torch.from_numpy(a).type(torch.FloatTensor)
torch.from_numpy(np.float32(a))
總結(jié)
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python數(shù)據(jù)分析庫(kù)pandas基本操作方法
下面小編就為大家分享一篇Python數(shù)據(jù)分析庫(kù)pandas基本操作方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-04-04從0編寫區(qū)塊鏈之用python解釋區(qū)塊鏈最基本原理
人工智能和區(qū)塊鏈誕生至今已經(jīng)有了十幾年,當(dāng)這些技術(shù)出現(xiàn)時(shí),人們都說他們會(huì)改變世界,但至今為止,這兩項(xiàng)技術(shù)對(duì)現(xiàn)實(shí)的影響依然有限,接下來通過本文介紹下從0編寫區(qū)塊鏈之用python解釋區(qū)塊鏈最基本原理,需要的朋友可以參考下2022-02-02淺談python數(shù)據(jù)結(jié)構(gòu)之動(dòng)態(tài)規(guī)劃
這篇文章主要介紹了淺談python數(shù)據(jù)結(jié)構(gòu)之動(dòng)態(tài)規(guī)劃,可能很多小伙伴會(huì)覺得這個(gè)詞很陌生,覺得這是一種很復(fù)雜的思想,學(xué)習(xí)起來很困難,其實(shí)并不是這樣,動(dòng)態(tài)規(guī)劃所講述的知識(shí)與動(dòng)態(tài)與規(guī)劃并無太大關(guān)聯(lián),需要的朋友可以參考下2023-07-07使用python快速獲取PDF文件頁(yè)數(shù)的辦法
有時(shí)在處理或打印一個(gè)PDF文檔之前,你可能需要先知道該文檔包含多少頁(yè),對(duì)于程序員來說,編寫腳本來完成這項(xiàng)工作會(huì)更加高效,本文就介紹一個(gè)使用Python快速獲取PDF文件頁(yè)數(shù)的辦法,需要的朋友可以參考下2024-03-03解決Python 寫文件報(bào)錯(cuò)TypeError的問題
這篇文章主要介紹了解決Python 寫文件報(bào)錯(cuò)TypeError的問題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-10-10詳解Python 多線程 Timer定時(shí)器/延遲執(zhí)行、Event事件
這篇文章主要介紹了Python 多線程 Timer定時(shí)器/延遲執(zhí)行、Event事件的相關(guān)知識(shí),本文給大家介紹的非常詳細(xì),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-06-06