tensorflow中tf.multinomial函数的使用方法讲解

tf.multinomial(logits, num_samples, seed=None, name=None)
logits 大小为[batch,n_class]
num_samples 表示采样的个数
seed 随机种子数
name表示该op的名字

这个函数就是根据logits中每个类别的概率采样,这个概率可以不用归一化后的概率,也就是每个类的概率可以大于1,不需要所有类别概率和为1,我觉得它底层是会自动给我们归一化的,输出结果为[batch,num_samples]大小
表示在每个batch_size 上采样num_samples个数,他是根据logits中的概率来采样的
这个函数用在rnn的输出中比较多,

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
`torch.multinomial()`函数是PyTorch的一个函数,用于从多项式分布抽取样本。多项式分布是一种离散概率分布,它描述了在一系列独立的重复试验,每个试验有多个可能的结果,每个结果发生的概率是固定的,且每个试验之间的结果是相互独立的。在深度学习,多项式分布通常用于对分类问题进行建模。 `torch.multinomial()`函数的语法如下: ```python torch.multinomial(input, num_samples, replacement=False, *, generator=None, out=None) -> LongTensor ``` 其,参数`input`是一个张量,表示多项式分布的概率分布。参数`num_samples`是一个整数,表示要抽取的样本数量。参数`replacement`是一个布尔值,表示是否进行有放回的抽样。如果为`True`,则进行有放回的抽样;如果为`False`,则进行无放回的抽样。参数`generator`是一个随机数生成器,用于生成随机数。参数`out`是一个输出张量,用于存储抽样结果。 以下是一个使用`torch.multinomial()`函数从多项式分布抽取样本的例子: ```python import torch # 定义一个包含五个类别的多项式分布 probs = torch.tensor([0.1, 0.2, 0.3, 0.25, 0.15]) # 将分布转化成二维形式 probs_2d = probs.view(1, -1) # 抽取三个样本 samples = torch.multinomial(probs_2d, 3) print(samples) # tensor([[2, 3, 2]]) ``` 上述代码,我们首先定义了一个包含五个类别的多项式分布,然后将其转化成二维形式。接着,我们使用`torch.multinomial()`函数抽取了三个样本,并将结果存储在`samples`张量。最后,我们输出了`samples`张量的值。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值