复制1.為什么選擇PyTorch
  1）簡潔：追求更少封裝，避免重復造輪子
  2）速度：靈活性不以速度為代價
  3）易用：語法優雅，簡單易學
  4）資源：社區豐富，維護及時

复制2.基礎入門
  1）安裝（pip install torch)
  2）創建tensor（PyTorch重要的數據結構）
  3）自動求導
  
import numpy as np  
import torch


#創建3行4列tensor，設置自動求導
x = torch.randn(3,4,requires_grad=True)
#輸出結構
x

复制#初始化b
b = torch.randn(3,4,requires_grad=True)
#計算t
t = x + b
#求和
y = t.sum()
#反向傳播計算
y.backward()
#輸出b梯度
b.grad

复制實戰一個例子（計算過程如下）：

复制#計算流程
x = torch.rand(1)
b = torch.rand(1, requires_grad = True)
w = torch.rand(1, requires_grad = True)
y = w * x 
z = y + b 
#如果不清空會累加起來
z.backward(retain_graph=True)
#w梯度
w.grad

复制#b梯度
b.grad

复制3.搭建線性回歸模型
  1）構造數據
  2）初始化參數
  3）前向傳播
  4）模型訓練
  5）預測結果
  6）保存模型

复制#初始化x值
x_values = [i for i in range(11)]
#轉換為ndarray
x_train = np.array(x_values, dtype=np.float32)
#維度轉換
x_train = x_train.reshape(-1, 1)
#查看形狀，（11，1）
x_train.shape


#計算y值
y_values = [2*i + 1 for i in x_values]
#轉換為ndarray
y_train = np.array(y_values, dtype=np.float32)
#維度轉換
y_train = y_train.reshape(-1, 1)
#查看形狀，（11，1）
y_train.shape

复制import torch
import torch.nn as nn


#線性回歸模型（本質是一個不加激活函數的全連接層）
class LinearRegressionModel(nn.Module):
    #初始化參數
    def __init__(self, input_dim, output_dim):
        super(LinearRegressionModel, self).__init__()
        self.linear = nn.Linear(input_dim, output_dim)  
    #前向傳播
    def forward(self, x):
        out = self.linear(x)
        return out


#輸入維度
input_dim = 1
#輸出維度
output_dim = 1
#初始化模型
model = LinearRegressionModel(input_dim, output_dim)
#輪數
epochs = 1000
#學習率
learning_rate = 0.01
#隨機梯度下降算法
optimizer = torch.optim.SGD(model.parameters(), 
                            lr=learning_rate)
#定義均方損失函數
criterion = nn.MSELoss()


#遍歷每一輪（這里使用CPU進行訓練，建議使用GPU速度快）
for epoch in range(epochs):
    #計算輪數
    epoch += 1
    #注意轉換成tensor
    inputs = torch.from_numpy(x_train)
    labels = torch.from_numpy(y_train)
    #梯度要清零每一次迭代
    optimizer.zero_grad() 
    #前向傳播
    outputs = model(inputs)
    #計算損失
    loss = criterion(outputs, labels)
    #返向傳播
    loss.backward()
    #更新權重參數
    optimizer.step()
    #每50輪輸出損失值
    if epoch % 50 == 0:
        print('epoch {}, loss {}'.format(epoch, loss.item()))

复制#預測訓練數據集
predicted = model(torch.from_numpy(x_train).requires_grad_()).data.numpy()
#輸出結果，跟我們真實y值幾乎沒有差別
predicted

复制#保存模型
torch.save(model.state_dict(), 'model.pkl')