Pytorch教程(十九)torch.cat与torch.stack的区别
生活随笔
收集整理的這篇文章主要介紹了
Pytorch教程(十九)torch.cat与torch.stack的区别
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
這一節,將分析拼接(concatenating)和疊加(stacking)張量的區別。
首先看三個例子,一個是pytorch,一個是TensorFlow,一個是numpy。
stack和cat張量之間的區別可以用一句話描述:
Concatenating joins a sequence of tensors along an existing axis, and Stacking joins a sequence of tensors along a new axis.
也就是說cat是在已有的軸上進行連接,而stack是創建一個新軸,進行連接。
在pytorch中,我們通過stack和cat進行操作:
總結
以上是生活随笔為你收集整理的Pytorch教程(十九)torch.cat与torch.stack的区别的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 和平谁愿意当我师傅我有点技术但不多
- 下一篇: sklearn下的ROC与AUC原理详解