PyTorch 是什么？

它是一個(gè)基于 Python 的科學(xué)計(jì)算包, 其主要是為了解決兩類場景:

• NumPy 的替代品, 以使用 GPU 的強(qiáng)大加速功能
• 一個(gè)深度學(xué)習(xí)研究平臺(tái), 提供最大的靈活性和速度

Tensors（張量）

Tensors 與 NumPy 的 ndarrays 非常相似, 除此之外還可以在 GPU 上使用張量來加速計(jì)算.

 

from __future__ import print_function import torch

構(gòu)建一個(gè) 5x3 的矩陣, 未初始化的:

x = torch.Tensor(5, 3) print(x)

構(gòu)建一個(gè)隨機(jī)初始化的矩陣:

x = torch.rand(5, 3) print(x)

獲得 size:

print(x.size())

操作

針對操作有許多語法. 在下面的例子中, 我們來看看加法運(yùn)算.

加法: 語法 1

y = torch.rand(5, 3) print(x + y)

加法: 語法 2

print(torch.add(x, y))

NumPy Bridge

將一個(gè) Torch Tensor 轉(zhuǎn)換為 NumPy 數(shù)組, 反之亦然.

Torch Tensor 和 NumPy 數(shù)組將會(huì)共享它們的實(shí)際的內(nèi)存位置, 改變一個(gè)另一個(gè)也會(huì)跟著改變.

轉(zhuǎn)換一個(gè) Torch Tensor 為 NumP

總結(jié)

以上是生活随笔為你收集整理的pytorch学习 入门篇（一）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。