pytorch中tensor.expand()和tensor.expand_as()函式詳解
阿新 • • 發佈:2020-01-09
tensor.expend()函式
>>> 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],[4]])
可以看出expand()函式括號裡面為變形後的size大小,而且原來的tensor和tensor.expand()是不共享記憶體的。
tensor.expand_as()函式
>>> b=torch.tensor([[2,[5,5]]) >>> print(b.size()) torch.Size([3,2]) >>> a.expand_as(b) tensor([[2,[4]])
可以看出,b和a.expand_as(b)的size是一樣大的。且是不共享記憶體的。
以上這篇pytorch中tensor.expand()和tensor.expand_as()函式詳解就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。