pytorch函数整理

Posted 刘文巾

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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之深度神经网络概念全面整理

pytorch之深度神经网络概念全面整理

pytorch之深度神经网络概念全面整理

[pytorch]pytorch loss function 总结

Pytorch学习笔记3.深度学习基础

一份整理 | PyTorch是什么,为何选择它