最新国产好看的视频,伊人天堂AV在线,国产Aaaaaa视频,蜜臀视频在线观看一区,人妻av色图,密臀久久久精品影片,青青视频免费观看毛片,久草在线观看视,国产三级精品色情在线

淺談pytorch和Numpy的區(qū)別以及相互轉(zhuǎn)換方法

 更新時(shí)間:2018年07月26日 14:33:33   作者:qq_34535410  
今天小編就為大家分享一篇淺談pytorch和Numpy的區(qū)別以及相互轉(zhuǎn)換方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

如下所示:

# -*- 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)文章

最新評(píng)論

三门峡市| 西贡区| 漳平市| 华池县| 灵宝市| 武宣县| 伊吾县| 苍梧县| 洪洞县| 株洲县| 定兴县| 巴东县| 连平县| 南汇区| 博爱县| 南岸区| 班玛县| 基隆市| 鄂尔多斯市| 玉山县| 搜索| 镇原县| 行唐县| 客服| 高雄县| 大港区| 寻甸| 亳州市| 林口县| 龙泉市| 财经| 中方县| 南康市| 安图县| 金秀| 伊金霍洛旗| 东兴市| 二手房| 上犹县| 林州市| 千阳县|