pytorch函数整理
生活随笔
收集整理的這篇文章主要介紹了
pytorch函数整理
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
| neg | Tensor里面的內容取反 |
| torch.multinomial() | torch.multinomial(input, num_samples,replacement=False, out=None) → LongTensor 對input的每一行做n_samples次取值,輸出的張量是每一次取值時input張量對應行的下標。 輸入是一個input張量,一個取樣數量,和一個布爾值replacement。 ????????????????input張量可以看成一個權重張量,每一個元素代表其在該行中的權重。如果有? ? ? ? ? ? 元素為0,那么在其他不為0的元素被取完之前,這個元素是不會被取到的。 ??????????????n_samples是每一行的取值次數,該值不能大于每一樣的元素數,否則會報錯。? ? ? ? ? ? ? ? ????????????????replacement指的是取樣時是否是有放回的取樣,True是有放回,False無放? ? ? ? ? ? ? ? 回。 |
參考文獻
torch.multinomial()理解_monchin的博客-CSDN博客
總結
以上是生活随笔為你收集整理的pytorch函数整理的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: pytorch笔记: 搭建Skip—gr
- 下一篇: 文巾解题 981. 基于时间的键值存储