1. Pytorch的安裝方法

1.1 Pytorch介紹

• Pytorch是一款facebook發布的深度學習框架，由于易用性，友好性，深受廣大用戶青睞.
• 官網：https://pytorch.org/get-started/locally/

1.2 Pytorch常見的安裝方法

• conda命令安裝

  • 帶GPU的安裝方法：conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
  • 不帶GPU的安裝步驟：conda install pytorch-cpu torchvision-cpu -c pytorch

• 使用Pycharm可視化安裝

  • 點擊File->Settings彈出工具包安裝按鈕：
    

  • 點擊上圖中的加號彈出Available package對話框，在文本中輸入torch

• 點擊如圖所示的Install Package和Specify version選擇指定版本進行安裝
  

2. 張量Tensor

2.1 *張量的概念

• 張量是一個統稱，其中包含很多類型.
• 0階張量：標量、常數，0-D Tensor
• 1階張量：向量，1-D Tensor
• 2階張量：矩陣，2-D Tensor
• 3階張量：3-D Tensor

2.2 張量的數據類型

• tensor中的數據類型非常多，常見類型如下：
  

3. Pytorch中tensor的常用方法

3.1 Pytorch中tensor的常用方法包括：

• 獲取tensor中的數據（當tensor中只有一個元素是可用）：tensor.item()
• 轉化為數組：tensor.numpy()
• 獲取tensor的形狀：tensor.size()
• 改變tensor的形狀：tensor.view()
• 獲取tensor的階數：tensor.dim()
• 獲取最大值：tensor.max()

3.2 步驟

步驟一：獲取tensor中的數據(當tensor中只有一個元素可用)：tensor.item()

import numpy as np a = torch.tensor(np.arange(1)) print(a.item())print("--------------------")

步驟二：tensor轉化為數組

torch.manual_seed(1) b = torch.rand(2,3) print(b.numpy())print("--------------------")

步驟三： 獲取tensor形狀

print(a.size()) print(b.size())print("--------------------")

步驟四：形狀改變

c = b.view(3,2) print(c)print("--------------------")

步驟五：獲取階數

# 5. 獲取階數 d = torch.rand(4,3,4) print(d.dim()) print("--------------------")

步驟六：獲取最大值

# 6. 獲取最大值 e = c.max() f = b.max() print(e,f)

總結

以上是生活随笔為你收集整理的十七、Pytorch的安装和使用的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。