Python torch.flatten()函數(shù)案例詳解
先看函數(shù)參數(shù)：
torch.flatten(input, start_dim=0, end_dim=-1)
input: 一個 tensor，即要被“推平”的 tensor。
start_dim: “推平”的起始維度。
end_dim: “推平”的結(jié)束維度。
首先如果按照 start_dim 和 end_dim 的默認(rèn)值，那么這個函數(shù)會把 input 推平成一個 shape 為 [n][n] 的tensor，其中 nn 即 input 中元素個數(shù)。
如果我們要自己設(shè)定起始維度和結(jié)束維度呢？
我們要先來看一下 tensor 中的 shape 是怎么樣的：
t = torch.tensor([[[1, 2, 2, 1],
[3, 4, 4, 3],
[1, 2, 3, 4]],
[[5, 6, 6, 5],
[7, 8, 8, 7],
[5, 6, 7, 8]]])
print(t, t.shape)
運(yùn)行結(jié)果：
tensor([[[1, 2, 2, 1],
[3, 4, 4, 3],
[1, 2, 3, 4]],
[[5, 6, 6, 5],
[7, 8, 8, 7],
[5, 6, 7, 8]]])
torch.Size([2, 3, 4])
我們可以看到，最外層的方括號內(nèi)含兩個元素，因此 shape 的第一個值是 2；類似地，第二層方括號里面含三個元素，shape 的第二個值就是 3；最內(nèi)層方括號里含四個元素，shape 的第二個值就是 4。
示例代碼：
x = torch.flatten(t, start_dim=1)
print(x, x.shape)
y = torch.flatten(t, start_dim=0, end_dim=1)
print(y, y.shape)
運(yùn)行結(jié)果：
tensor([[1, 2, 2, 1, 3, 4, 4, 3, 1, 2, 3, 4],
[5, 6, 6, 5, 7, 8, 8, 7, 5, 6, 7, 8]])
torch.Size([2, 12])
tensor([[1, 2, 2, 1],
[3, 4, 4, 3],
[1, 2, 3, 4],
[5, 6, 6, 5],
[7, 8, 8, 7],
[5, 6, 7, 8]])
torch.Size([6, 4])
可以看到，當(dāng) start_dim = 11 而 end_dim = −1−1 時，它把第 11 個維度到最后一個維度全部推平合并了。而當(dāng) start_dim = 00 而 end_dim = 11 時，它把第 00 個維度到第 11 個維度全部推平合并了。pytorch中的 torch.nn.Flatten 類和 torch.Tensor.flatten 方法其實(shí)都是基于上面的 torch.flatten 函數(shù)實(shí)現(xiàn)的。
到此這篇關(guān)于Python torch.flatten()函數(shù)案例詳解的文章就介紹到這了,更多相關(guān)Python torch.flatten()函數(shù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Python利用re模塊實(shí)現(xiàn)簡易分詞(tokenization)
分詞（tokenization）任務(wù)是Python字符串處理中最為常見任務(wù)了。本文將利用re模塊實(shí)現(xiàn)簡易tokenizer，文中的示例代碼講解詳細(xì)，需要的可以參考一下2022-04-04
python實(shí)現(xiàn)超簡單的視頻對象提取功能
這篇文章主要給大家介紹了關(guān)于利用python實(shí)現(xiàn)超簡單的視頻對象提取功能的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2018-06-06
不知道什么時候開始，中國出現(xiàn)了南抖音、北快手的互文格局。喜歡抖音主要是兩個初衷，學(xué)做菜聽音樂。抖音捧紅了很多人，也讓很多本不怎么讓大家熟知的歌曲、BGM，經(jīng)過翻唱、混剪與視頻搭配，從而傳播大街小巷。有沒有想過將這些好聽的剪輯批量下載下來呢？2021-05-05
由于很多論文里面的測試圖片沒有標(biāo)號,就不能確定它們是Testset數(shù)據(jù)集中哪幾張圖片.為了能解決這個問題,需要完成以圖片去搜索整個數(shù)據(jù)集文件目錄的任務(wù).本文有非常詳細(xì)的圖文示例,需要的朋友可以參考下2021-05-05
獲取django框架orm query執(zhí)行的sql語句實(shí)現(xiàn)方法分析
這篇文章主要介紹了獲取django框架orm query執(zhí)行的sql語句實(shí)現(xiàn)方法,結(jié)合實(shí)例形式分析了Django框架中orm query執(zhí)行的sql語句獲取方法相關(guān)實(shí)現(xiàn)技巧,需要的朋友可以參考下2019-06-06
大家好，本篇文章主要講的是Python?selenium把歌詞評論做成詞云圖，感興趣的同學(xué)趕快來看一看吧，對你有幫助的話記得收藏一下2022-01-01
這篇文章主要介紹了python 如何實(shí)現(xiàn)一個截屏工具，幫助大家更好的理解和學(xué)習(xí)使用python，感興趣的朋友可以了解下2021-05-05
本篇文章主要介紹了python列表生成式與列表生成器的使用，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-02-02
Opencv實(shí)現(xiàn)二維直方圖的計(jì)算及繪制
這篇博客將介紹如何使用Opencv進(jìn)行二維直方圖的計(jì)算及繪制，維直方圖可以讓我們對不同的像素密度有更好的了解，感興趣的可以了解一下2021-07-07
pandas中DataFrame數(shù)據(jù)合并連接（merge、join、concat)
這篇文章主要給大家介紹了關(guān)于pandas中DataFrame 數(shù)據(jù)合并連接（merge、join、concat)的相關(guān)資料，文中介紹的非常詳細(xì)，需要的朋友可以參考下2021-05-05