首頁 > 軟體

Python檢視Tensor尺寸及檢視資料型別的實現

2022-07-15 10:01:15

檢視Tensor尺寸及檢視資料型別

Tensor尺寸檢視

命令:

  • x.shape

例子:

input = torch.randn(20,16,50,32)
input.shape

輸出為:

注意呼叫的時候不要用x.shape()

否則會報錯:AttributeError: 'Tensor' object has no attribute 'get_shape'

資料型別檢視

命令:

  • x.type()

例子:

input = torch.randn(20,16,50,32)
input.type()

輸出:

Pytorch基本資料型別tensor

在Pytorch中必須使用Pytorch特有的張量(tensor)資料型別,本文介紹tensor的基本操作

Python和Pytorch資料型別對應

以上資料是儲存在CPU中。

如果使用.cuda(),會返回一個GPU上的參照

import torch
a = torch.tensor([1,2])
c = a.cuda
print(c)
print(type(c))
-----輸出------
D:UsersLenovoanaconda3python.exe C:/Users/Lenovo/Desktop/pythonProject2/main.py
<built-in method cuda of Tensor object at 0x000002A03434CA00>
<class 'builtin_function_or_method'>
Process finished with exit code 0

建立tensor的方法

import torch
import numpy
a = numpy.array([1,2,3,4])
b = torch.from_numpy(a)  #從numpy中引入
print(b)
c = torch.tensor([1,2,3,4])
print(c)  #直接建立
d = torch.Tensor([4,5,6,7])
print(d)  #直接建立
e = torch.Tensor(2,3)
print(e)  #用shape建立隨機的指定維度的tensor

一些常用的生成tensor方法

import torch
a = torch.rand(3, 3) #生成指定大小的,元素範圍[0,1]的tensor
print(a)
b = torch.rand_like(a) #生成與物件tensor大小一致的tensor
print(b)
c = torch.randint(1,10,(3,3)) #生成(3,3)大小,[1,10)範圍的tensor(包括1,但不包括10)
print(c)
d = torch.randn(2,4) #生成均值為0,方差為1的隨機tensor
print(d)
e = torch.full([2,5],0) #生成全部一樣的tensor
print(e)
f = torch.arange(0,10)  #生成順序tensor
print(f)
g = torch.linspace(0,9,steps=8)  #生成[0,9],等分成8個的tensor
print(g)
print(torch.ones(3,1),
      torch.zeros(4,5),
      torch.eye(6)) #生成全1,全0,單位矩陣tensor

tensor的切片與索引

import torch
a = torch.rand(3, 7) #生成指定大小的,元素範圍[0,1]的tensor
print(a)
print("*"*100)
print(a[0])  #取一行
print("*"*100)
print(a[:2,4:])  #指定一塊子區域
print("*"*100)
print(a[:,2])  #取一列
print("*"*100)
print(a[:,0:7:2]) #[0,7]行隔2行取樣
print("*"*100)
print(a.index_select(1,torch.tensor([2,6])))  #指定切片位置
print("*"*100)
mask = a.ge(0.8)
print(torch.masked_select(a,mask)) #通過掩碼條件切片(注意切片後會被flatten)
print(mask) #看看mask矩陣

tensor的維度變換(重點)

基本操作:

  • view/reshape
  • squeeze/unsqueeze
import torch
a = torch.rand(3,4,2) #生成指定大小的,元素範圍[0,1]的tensor
print(a)
# view方法:變換tensor的形狀
print(a.view(3, 8))  #壓縮維度
print(a.view(3,2,2,2))  #擴充套件維度
# unsqueeze方法:在指定地方插入一維(squeeze方法同理會消除一維)
print(a.unsqueeze(1))
print(a.squeeze(1))  #squeeze只有在原有維度為1才有用,試試把上面改成rand(3,1,2)看看效果

tensor的疊加和分割

```python
import torch
#cat操作
a = torch.rand(4,1,3)
b = torch.rand(3,1,3)
# print(a)
# print(b)
c = torch.cat((a,b))  #合併tensor(只能在其他dimension一致的情況下才能合併)
# print(c)
# print(c.shape)
#stack操作
d = torch.rand(4,1)
e = torch.rand(4,1)
# print(d)
# print(e)
f = torch.stack((d,e))  #合併tensor,與cat不同的是,stack會增加一個更高的維度
# print(f)
# print(f.shape)
#split操作
g = torch.rand(5,2,1)
h,i = g.split([1,4])
# print(g)
# print(h)
# print(i)

tensor的數學運算

import torch
#基本加減乘除
a = torch.zeros(4,3)
b = torch.ones(3)
# print(a+b)  #這樣會報錯,正確做法如下
# print(torch.add(a, b))  #結果全是1,broadcast運算邏輯
c = torch.tensor([[1,2,3],[4,5,6],[7,8,9]])
d = torch.eye(3)
# print(torch.add(c,d))
# print(c+d)  #不採用broadcast邏輯時,這樣也可以
# print(c*d)
# print(c/d)
# print(c**2) #平方
# print(c**0.5) #開平方
#矩陣相乘
# print(torch.matmul(torch.ones(3,3),torch.ones(3,3)))
# print(torch.ones(3,3)@torch.ones(3,3))  #用@的效果是一樣的
#高維度的矩陣相乘
e = torch.rand(4,3,2,3)
f = torch.rand(4,3,3,5)
g = e@f
# print(g)
# print(g.shape)  #可以看到,本質上也是二維矩陣相乘的規律
#clamp演演算法
print(c.clamp(4))  #把小於4的值全部都替換成4

tensor的統計相關操作

import torch
#norm方法(求範數)
a = torch.arange(10,dtype=float)
b = a.view(2,5)
print(a.norm(1))
print(b.norm(1))
print(a.norm(2,dtype=float))  #同理求二範數
print(b)
print(b.norm(1,dim=1,dtype=float))  #求指定維度的範數
#求最大、最小、平均、求和
print(a.sum())
print(a.min())
print(a.max())
print(a.mean())
#top N的值
c = torch.tensor([1,2,3,3,4,4,4,5,5,5,5,6,8,8],dtype=float)
print(c.topk(3))
print(c.topk(3, largest=False))  #找到前N最小的值
print(c.kthvalue(4))  #找到第k小的值

#比較
print(a>4)
print(a!=8)
#where
cond = torch.tensor([[1,2],[3,4]],dtype=float)  #用where組合2個tensor
d = torch.zeros(2,2)
e = torch.ones(2,2)
print(torch.where(cond>2,d,e))

以上為個人經驗,希望能給大家一個參考,也希望大家多多支援it145.com。


IT145.com E-mail:sddin#qq.com