日韩性视频-久久久蜜桃-www中文字幕-在线中文字幕av-亚洲欧美一区二区三区四区-撸久久-香蕉视频一区-久久无码精品丰满人妻-国产高潮av-激情福利社-日韩av网址大全-国产精品久久999-日本五十路在线-性欧美在线-久久99精品波多结衣一区-男女午夜免费视频-黑人极品ⅴideos精品欧美棵-人人妻人人澡人人爽精品欧美一区-日韩一区在线看-欧美a级在线免费观看

歡迎訪問 生活随笔!

生活随笔

當(dāng)前位置: 首頁 > 编程资源 > 编程问答 >内容正文

编程问答

Pytorch 笔记——tensor

發(fā)布時(shí)間:2025/4/5 编程问答 22 豆豆
生活随笔 收集整理的這篇文章主要介紹了 Pytorch 笔记——tensor 小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.

tensor和numpy中的ndarray很相似,唯一的不同是tensor可以應(yīng)用在GPU上,來加速計(jì)算。

1?創(chuàng)建tensor

randn(size)

創(chuàng)建一個(gè)標(biāo)準(zhǔn)正態(tài)分布內(nèi)隨機(jī)的矩陣

zeros(size)

全零矩陣

tensor(list)

創(chuàng)建一個(gè)內(nèi)容為參數(shù)的tensor

linspace

同numpy

randperm

返回一個(gè)0到n-1的數(shù)組

基本上創(chuàng)建的這些函數(shù)都有一個(gè)對(duì)應(yīng)的like函數(shù)(randn_like,zeros_like…)來創(chuàng)建一個(gè)size和形參一樣的tensor

2?tensor和ndarray的轉(zhuǎn)換

無論誰轉(zhuǎn)換成誰,轉(zhuǎn)換的tensor和ndarray都是另一個(gè)的引用,要變一起變。

2.1?tensor->ndarray

2.2?ndarray->tensor

3?cuda和cpu的tensor切換

4 一些關(guān)于tensor的運(yùn)算

很多運(yùn)算符都在末尾有一個(gè)_,表示inplace操作

加法

+

tensor.add

torch.add

修改一個(gè)的值

(很多運(yùn)算符都在末尾有一個(gè)_,表示inplace操作

5. torch的函數(shù)

很多函數(shù) torch.func(tensor)和tensor.func()是等價(jià)的。

size()

tensor的形狀

view

torch里面的“reshape”

item

如果tensor中只有一個(gè)元素的時(shí)候,使用這個(gè)函數(shù)獲取這個(gè)

數(shù)值

abs

絕對(duì)值

sin

mm

矩陣相乘

pow

cat

合并數(shù)組

axis=0 沿著列的方向[1…n][0]

axis=1 沿著行的方向[0][1;;;n]

axis=0 (從左往右數(shù))第0個(gè)坐標(biāo)的元素疊加,其他坐標(biāo)的元素不變

axis=1 第1個(gè)坐標(biāo)的元素疊加,其他坐標(biāo)的元素不變

max

如果又axis參數(shù)的話,返回的第一個(gè)下標(biāo)是所有的最大值,第二個(gè)下標(biāo)是最大值對(duì)應(yīng)的index。

axis=0 [1~n][0]里面找到最大的一個(gè),

也就是第零個(gè)坐標(biāo)的參數(shù)變化(去比較),其他坐標(biāo)的參數(shù)不變

axis=1 [0][1~n]里面找到最大的一個(gè),

也就是第一個(gè)坐標(biāo)的參數(shù)變化(去比較),其他坐標(biāo)的參數(shù)不變

narrow()

截取tensor的一部分 torch.narrow(input, dim, start, length)

?返回的部分和input 共享內(nèi)存

?

input 需要被截取的tensor

dim 沿著那個(gè)維度截取

start 起始截取下標(biāo)

length? 截取多長的

?

(沿著第0個(gè)下標(biāo)截取)?

?

?

(沿著第1個(gè)下標(biāo)截取)

squeeze()

我們首先創(chuàng)建一個(gè)維度為(1,2,3)的tensor

然后對(duì)他squeeze

只有squeeze對(duì)應(yīng)的維度為1的時(shí)候,我們才會(huì)把這個(gè)維度去掉,不然的話維度保持不變

unsqueeze

masked_fill(data,x)

將data為True的相應(yīng)位置的值替換為x

?numel

tensor中元素的個(gè)數(shù)

permute?

tensor維度換位

(2,0,1)的意思是,維度變換成:(原來的第二個(gè),原來的第0個(gè),原來的第1個(gè))?

t

轉(zhuǎn)置
logical_not邏輯取反

總結(jié)

以上是生活随笔為你收集整理的Pytorch 笔记——tensor的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò),歡迎將生活随笔推薦給好友。