建立、檢視形狀、建立指定形式的張量、操作方法（加減乘除）以及操作裝置（cpu/gpu）
1）torch.tesor()建立張量
2）torch.view()對張量進行降維
3）torch.size()檢視張量的形狀
4）torch.ones() torch.zeros()建立指定形式的張量
5）torch.to(device)
在使用torch之前，要對其進行匯入
import torch
建立乙個torch
1 x = torch.tensor([12, 5])2print(x)
得到torch的維度
print(x.size())
torch的resize
1#resizing
2 x = torch.ones(3, 3)
3print
(x)4 y = x.view(9)
5print
(y)6 z = x.view(-1)
7print(z)
1 tensor([[1., 1., 1.],2 [1., 1., 1.],
3 [1., 1., 1.]])
4 tensor([1., 1., 1., 1., 1., 1., 1., 1., 1.])
5 tensor([1., 1., 1., 1., 1., 1., 1., 1., 1.])
同樣也可以使用x.size(-1)實現
使用gpu進行運算
1 device = torch.device('
cuda
')
1 x = torch.ones(5, 5, device=device)
或者將定義好的torch遷移到gpu上
1 y = torch.ones(5, 5, dtype=torch.float)2 y = y.to(device)
同樣的，也可以遷移回cpu
1 x = x.to('
cpu'
)2 y = y.to('
cpu')
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入門1 1（基本操作）
本節繼續來講pytorch的入門級操作，加加減減 拍平拉直的操作。在操作張量的的時候，肯定少不了張量之間的加減乘除。二維世界的張量就是矩陣，是不是像我們學數學那樣直接用 直接操作就好了呢？x torch.tensor 1,2 3,4 y torch.ones 2,2 x y 對應元素相加 x y 對...
pytorch入門1 0（基本操作）
pytorch入門1.0 主要是關於張量的建立 運算 索引等一些基本操作。隨便練習一下，增加對張量操作的熟悉程度。1.pytorch是什麼？pytorch是2017年由facebook人工智慧研究院 fair 基於torch推出的乙個開源python機器學習庫。該庫能借助gpu加速張量的計算 亦具有...