pytorch中tensor.expand()和tensor.expand_as()函數(shù)詳解
tensor.expend()函數(shù)
>>> import torch >>> a=torch.tensor([[2],[3],[4]]) >>> print(a.size()) torch.Size([3, 1]) >>> a.expand(3,2) tensor([[2, 2], [3, 3], [4, 4]]) >>> a tensor([[2], [3], [4]])
可以看出expand()函數(shù)括號里面為變形后的size大小,而且原來的tensor和tensor.expand()是不共享內(nèi)存的。
tensor.expand_as()函數(shù)
>>> b=torch.tensor([[2,2],[3,3],[5,5]]) >>> print(b.size()) torch.Size([3, 2]) >>> a.expand_as(b) tensor([[2, 2], [3, 3], [4, 4]]) >>> a tensor([[2], [3], [4]])
可以看出,b和a.expand_as(b)的size是一樣大的。且是不共享內(nèi)存的。
以上這篇pytorch中tensor.expand()和tensor.expand_as()函數(shù)詳解就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
Python利用Diagrams繪制漂亮的系統(tǒng)架構圖
Diagrams 是一個基于Python繪制云系統(tǒng)架構的模塊,它能夠通過非常簡單的描述就能可視化架構。本文將利用它繪制漂亮的系統(tǒng)架構圖,感興趣的可以了解一下2023-01-01基于python+pandoc實現(xiàn)html批量轉(zhuǎn)word
pandoc是一個強大的文檔格式轉(zhuǎn)換工具,支持豐富的格式轉(zhuǎn)換,并盡可能的保留原來的排版,號稱文檔格式轉(zhuǎn)換的瑞士軍刀,本文將給大家介紹一下使用python搭配pandoc實現(xiàn)html批量轉(zhuǎn)word,感興趣的朋友可以參考閱讀下2023-09-09Python繪圖之自定義圖類型控件實現(xiàn)混合類型圖表
這篇文章主要為大家詳細介紹了Python如何新建繪圖類型控件,實現(xiàn)混合類型圖表,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下2023-08-08Python web框架(django,flask)實現(xiàn)mysql數(shù)據(jù)庫讀寫分離的示例
這篇文章主要介紹了Python web框架(django,flask)實現(xiàn)mysql數(shù)據(jù)庫讀寫分離的示例,幫助大家更好的理解和學習python,感興趣的朋友可以了解下2020-11-11Python開發(fā)游戲之井字游戲的實戰(zhàn)步驟
最近正在學習Python,所以最近做了一個關于Python的實例,下面這篇文章主要給大家介紹了關于Python開發(fā)游戲之井字游戲的相關資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下2023-02-02