import torch
#1
x = torch.randn(2, 1, 1)#為1可以擴展為3和4
x = x.expand(2, 3, 4)
print('x :', x.size())
>>> x : torch.Size([2, 3, 4])
#2
#擴展一個新的維度必須在最前面，否則會報錯
x = x.expand(2, 3, 4, 6)
>>> RuntimeError: The expanded size of the tensor (3) must match the existing size (2) at non-singleton dimension 1.
x = x.expand(6, 2, 3, 4)
>>> x : torch.Size([6, 2, 3, 4])
#3
#某一個維度為-1表示不改變該維度的大小
x = x.expand(6, -1, -1, -1)
>>> x : torch.Size([6, 2, 1, 1])

import torch
#1
x = torch.randn(2, 1, 1)#原維度為1可以擴展為其他維度
y = torch.randn(2, 3, 3)
x = x.expand_as(y)
print('x :', x.size())
>>> x : torch.Size([2, 3, 3])
#2
x = torch.randn(2, 2, 2)#原維度為其他不是1的值不可以擴展為其他維度
y = torch.randn(2, 3, 4)
x = x.expand_as(y)
print('x :', x.size())
>>> RuntimeError: The expanded size of the tensor (4) must match the existing size (2) at non-singleton dimension 2.  Target sizes: [2, 3, 4].

Pytorch expand()函數(shù)

返回tensor的一個新視圖

單個維度擴大為更大的尺寸。

tensor也可以擴大為更高維，新增加的維度將附在前面。

擴大tensor不需要分配新內(nèi)存，只是僅僅新建一個tensor的視圖，其中通過將stride設(shè)為0，一維將會擴展位更高維。

任何一個一維的在不分配新內(nèi)存情況下可擴展為任意的數(shù)值。

note:使用expand()函數(shù)的時候

x自身不會改變，因此需要將結(jié)果重新賦值。

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python編程中內(nèi)置的NotImplemented類型的用法
這篇文章主要介紹了Python編程中內(nèi)置的NotImplemented類型的用法，NotImplemented 是Python在內(nèi)置命名空間中的六個常數(shù)之一，下文更多詳細內(nèi)容需要的小伙伴可以參考一下
2022-03-03
Python使用pyautocad+openpyxl處理cad文件示例
這篇文章主要介紹了Python使用pyautocad+openpyxl處理cad文件,結(jié)合實例形式分析了Python使用pyautocad與openpyxl模塊讀寫cad文件相關(guān)應(yīng)用操作技巧,需要的朋友可以參考下
2019-07-07
Python實現(xiàn)強制復(fù)制粘貼的示例詳解
下個文檔還要馬內(nèi)？還好我會Python，本文就來教大家來一手如何利用Python實現(xiàn)強制復(fù)制粘貼。文中的示例代碼講解詳細，感興趣的小伙伴可以了解一下
2022-12-12
Python中的pickle模塊解析
這篇文章主要介紹了Python中的pickle模塊解析,pickle 模塊和 json 模塊很像,都有序列化的功能,不過 pickle 模塊更加局限一些只能對 python 使用,它可以對一個 python 對象結(jié)構(gòu)的二進制序列化和反序列化,需要的朋友可以參考下
2023-09-09
pandas?如何將字符串映射為數(shù)字
這篇文章主要介紹了pandas?如何將字符串映射為數(shù)字，具有很好的參考價值，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2022-07-07
使用Python實現(xiàn)NBA球員數(shù)據(jù)查詢小程序功能
這篇文章主要介紹了使用Python實現(xiàn)NBA球員數(shù)據(jù)查詢小程序功能,本文通過實例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2020-11-11
python手機號前7位歸屬地爬蟲代碼實例
這篇文章主要介紹了python手機號前7位歸屬地爬蟲代碼實例,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
2020-03-03
如何定義TensorFlow輸入節(jié)點
今天小編就為大家分享一篇如何定義TensorFlow輸入節(jié)點，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-01-01
Python使用grequests并發(fā)發(fā)送請求的示例
這篇文章主要介紹了Python使用grequests并發(fā)送請求的示例，幫助大家更好的理解和使用python，感興趣的朋友可以了解下
2020-11-11
Github?Copilot的申請以及在Pycharm的配置與使用詳解
GitHub在聯(lián)合OpenAI推出了一款"GitHub Copilot"工具,可以根據(jù)上下文自動寫代碼,下面這篇文章主要給大家介紹了關(guān)于Github?Copilot的申請以及在Pycharm的配置與使用的相關(guān)資料,文中通過圖文以及實例代碼介紹的非常詳細,需要的朋友可以參考下
2022-04-04