pytorch入門1 0（基本操作）

pytorch入門1.0

主要是關於張量的建立、運算、索引等一些基本操作。隨便練習一下，增加對張量操作的熟悉程度。

1.pytorch是什麼？

pytorch是2023年由facebook人工智慧研究院（fair）基於torch推出的乙個開源python機器學習庫。該庫能借助gpu加速張量的計算；亦具有自動求導系統。

2.使用pytorch的前期工作

首先得安裝好；（假定你已經安裝好啦）

匯入torch包；

import torch

print(torch.__version__) # 列印pytorch的版本
# 設定隨機種子，保證在神經網路中的各種權重的初始化一致。
torch.manual_seed(42)
torch.cuda.manual_seed(42)

# 建立空的張量（值為垃圾值）

torch.empty(1)
torch.empty(3).dtype
# 建立張量（值為垃圾值）
torch.tensor(1)
torch.tensor(1).dtype
#依照元素值建立張量，更像是型別轉換
torch.tensor([1.0,3.0])
torch.tensor([1.0,3.0]).dtype
# 建立二維矩陣，對角線元素為1，其他位置元素為0
torch.eye(3,dtype=torch.float32)
# 建立乙個形狀跟引數裡的tensor一樣的張量
torch.empty_like(torch.tensor(3,3))
# 建立乙個全是1的向量，例如2*3*4
torch.ones(2,3,4,device='cuda') # 把此變數放到gpu上，devive='cuda'
# 建立乙個全0向量
torch.zeros(2,4)
# 建立乙個3*3*3的張量，每個元素從2到5[2,5)隨機取值
torch.randint(2,5,(3,3,3))

pytorch入門1 1（基本操作）

本節繼續來講pytorch的入門級操作，加加減減 拍平拉直的操作。在操作張量的的時候，肯定少不了張量之間的加減乘除。二維世界的張量就是矩陣，是不是像我們學數學那樣直接用 直接操作就好了呢？x torch.tensor 1,2 3,4 y torch.ones 2,2 x y 對應元素相加 x y 對...

pytorch基本操作

coding utf 8 import torch import numpy as np 根據torch.tensor生成張量 print torch.tensor 1 print torch.tensor 2,3 print torch.tensor 2,3 根據torch.tensor生成張量 ...

Pytorch的基本操作

建立 檢視形狀 建立指定形式的張量 操作方法 加減乘除 以及操作裝置 cpu gpu 1 torch.tesor 建立張量 2 torch.view 對張量進行降維 3 torch.size 檢視張量的形狀 4 torch.ones torch.zeros 建立指定形式的張量 5 torch.to ...