淺談pytorch和Numpy的區(qū)別以及相互轉(zhuǎn)換方法
如下所示:
# -*- coding: utf-8 -*- # @Time : 2018/1/17 16:37 # @Author : Zhiwei Zhong # @Site : # @File : Numpy_Pytorch.py # @Software: PyCharm import torch import numpy as np np_data = np.arange(6).reshape((2, 3)) # numpy 轉(zhuǎn)為 pytorch格式 torch_data = torch.from_numpy(np_data) print( '\n numpy', np_data, '\n torch', torch_data, ) ''' numpy [[0 1 2] [3 4 5]] torch 0 1 2 3 4 5 [torch.LongTensor of size 2x3] ''' # torch 轉(zhuǎn)為numpy tensor2array = torch_data.numpy() print(tensor2array) """ [[0 1 2] [3 4 5]] """ # 運(yùn)算符 # abs 、 add 、和numpy類(lèi)似 data = [[1, 2], [3, 4]] tensor = torch.FloatTensor(data) # 轉(zhuǎn)為32位浮點(diǎn)數(shù),torch接受的都是Tensor的形式,所以運(yùn)算前先轉(zhuǎn)化為T(mén)ensor print( '\n numpy', np.matmul(data, data), '\n torch', torch.mm(tensor, tensor) # torch.dot()是點(diǎn)乘 ) ''' numpy [[ 7 10] [15 22]] torch 7 10 15 22 [torch.FloatTensor of size 2x2] '''
以上這篇淺談pytorch和Numpy的區(qū)別以及相互轉(zhuǎn)換方法就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
ROS1?rosbag的詳細(xì)使用并且使用python合并bag包的方法
這篇文章主要介紹了ROS1?rosbag的詳細(xì)使用,并且使用python來(lái)合并bag包,本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2023-05-05
python 實(shí)現(xiàn)超級(jí)瑪麗游戲
這篇文章主要介紹了python 實(shí)現(xiàn)超級(jí)瑪麗游戲,幫助大家更好的理解和使用python開(kāi)發(fā)游戲,感興趣的朋友可以了解下2020-11-11
python 爬蟲(chóng)出現(xiàn)403禁止訪問(wèn)錯(cuò)誤詳解
這篇文章主要介紹了 python 爬蟲(chóng)解決403禁止訪問(wèn)錯(cuò)誤的相關(guān)資料,需要的朋友可以參考下2017-03-03
Python如何在ubuntu中更改Python和pip指向
這篇文章主要介紹了Python如何在ubuntu中更改Python和pip指向問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-08-08
python 定義n個(gè)變量方法 (變量聲明自動(dòng)化)
今天小編就為大家分享一篇python 定義n個(gè)變量方法 (變量聲明自動(dòng)化),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-11-11
python模擬鍵盤(pán)輸入 切換鍵盤(pán)布局過(guò)程解析
這篇文章主要介紹了python模擬鍵盤(pán)輸入 切換鍵盤(pán)布局過(guò)程解析,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-08-08
Python結(jié)合Selenium簡(jiǎn)單實(shí)現(xiàn)Web自動(dòng)化測(cè)試
這篇文章是入門(mén)級(jí)別的應(yīng)用Python + Selenium進(jìn)行自動(dòng)化測(cè)試,包括環(huán)境搭建及簡(jiǎn)單的實(shí)例,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2021-09-09
python自動(dòng)保存百度盤(pán)資源到百度盤(pán)中的實(shí)例代碼
這篇文章主要介紹了python自動(dòng)保存百度盤(pán)資源到百度盤(pán)中的實(shí)例代碼,代碼簡(jiǎn)單易懂,非常不錯(cuò),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-08-08
Python3.5實(shí)現(xiàn)的三級(jí)菜單功能示例
這篇文章主要介紹了Python3.5實(shí)現(xiàn)的三級(jí)菜單功能,涉及Python針對(duì)json格式數(shù)據(jù)的讀取、遍歷、查找、判斷等相關(guān)操作技巧,需要的朋友可以參考下2019-03-03

