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无放 回。 |
参考文献
以上是关于pytorch函数整理的主要内容,如果未能解决你的问题,请参考以下文章