Pytorch 快速入門搭建網絡

Pytorch簡介

 

Pytorch快速安裝

1.先提前安裝後python環境，C++/Java也可，本文以Python爲例。

2.進入Pytorch官網：https://pytorch.org/

3.往下拉找到下面的界面，根據自己的實際情況選擇配置。（我這裏選擇用pip安裝）

4.然後講Run this Command中生成的代碼拷貝下來。

5.進入windows的終端（cmd），複製並執行拷貝的命令，等待下載並安裝完成（我這個版本一共899兆，比較大，大家耐心等待）。

安裝完畢：

6.測試：新建python文件，輸入如下代碼

import torch
import torchvision

print(torch)

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

輸出如下，如果可以正常輸出（兩個print依次對應圖中的兩個輸出），則安裝完畢。

 

Pytorch 常用語法

1.張量的使用（可以理解爲矩陣）

# -- 構造矩陣
x = torch.empty(5,3)   # 構造一個5行3列的矩陣，並不初始化
print(x)

print("")
x = torch.rand(5,3)     # 隨機初始化矩陣
print(x)

print("")
x = torch.zeros(5,3, dtype=torch.long)     # 初始化0矩陣，類型爲long，dtype=torch.long可以省略，也可以是int，等等
print(x)

print("")
x = torch.ones(4,4)      # 初始化1矩陣
print(x)

print("")
x = torch.tensor([[5.5, 3],[5.5, 3]])      # 直接用數據初始化
print(x)


# -- 矩陣運算

x = torch.zeros(3,3)        # 矩陣加法
y = torch.ones(3,3)
z = torch.add(x,y)
print(z)
z = torch.add(y,z)
print(z)
torch.add(y,z,out=x)        # 指定輸出參數
print(x)
x.add_(z)                   # x = x + z
print(x)
x.add(z)                   # x + z, 沒有賦值給x的過程
print(x)

print(x[:,1])              #輸出一列

2.改變矩陣形狀，獲取矩陣的size

import torch
import torchvision

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

y = x.view(18)
print(y)
print(y.size())

y = x.view(9,2)
print(y)
print(y.size())

y = x.view(2,9)
print(y)
print(y.size())

y = x.view(-1,3)        # -1 表示自動匹配
print(y)
print(y.size())

z = y.size(0)
print(z)

3.後使用過程中遇到再補充

 

Pytroch 快速搭建一個神經網絡