生成/logits_采样器

• 生成/logits_采样器
  • 静态
    • .LogitsSampler
      • new LogitsSampler(generation_config)
      • 实例
        • ._call(logits) ⇒ *
        • .sample(logits) ⇒ *
        • .getLogits(logits, index) ⇒ Float32Array
        • .randomSelect(probabilities) ⇒ number
      • 静态
        • .getSampler(generation_config) ⇒ LogitsSampler
  • 内部
    • ~GreedySampler
      • .sample(logits) ⇒ *
    • ~MultinomialSampler
      • .sample(logits) ⇒ *
    • ~BeamSearchSampler
      • .sample(logits) ⇒ *

---

generation/logits_sampler.LogitsSampler

采样器是用于文本生成的所有采样方法的基类。

类型：generation/logits_sampler 的静态类

• .LogitsSampler
  • new LogitsSampler(generation_config)
  • 实例
    • ._call(logits) ⇒ *
    • .sample(logits) ⇒ *
    • .getLogits(logits, index) ⇒ Float32Array
    • .randomSelect(probabilities) ⇒ number
  • 静态
    • .getSampler(generation_config) ⇒ LogitsSampler

---

new LogitsSampler(generation_config)

使用指定的生成配置创建一个新的 Sampler 对象。

参数	类型	描述
generation_config	GenerationConfig	生成配置。

---

logitsSampler._call(logits) ⇒ <code> * </code>

执行采样器，使用指定的 logits。

类型：LogitsSampler 的实例方法

参数	类型
logits	张量

---

logitsSampler.sample(logits) ⇒ <code> * </code>

用于采样 logits 的抽象方法。

类型：LogitsSampler 的实例方法
抛出:

• Error 如果在子类中未实现。

参数	类型
logits	张量

---

logitsSampler.getLogits(logits, index) ⇒ <code>Float32Array</code>

返回指定的 logits 作为数组，并应用温度。

类型：LogitsSampler 的实例方法

参数	类型
logits	张量
index	数字

---

logitsSampler.randomSelect(probabilities) ⇒ <code>number</code>

根据指定的概率随机选择一个项目。

类型：LogitsSampler 的实例方法
返回值：number - 所选项目的索引。

参数	类型	描述
probabilities	*	用于选择的概率数组。

---

LogitsSampler.getSampler(generation_config) ⇒ <code>LogitsSampler</code>

根据指定的选项返回一个 Sampler 对象。

类型：LogitsSampler 的静态方法
返回值：LogitsSampler - 一个 Sampler 对象。

参数	类型	描述
generation_config	GenerationConfig	包含采样器选项的对象。

---

generation/logits_sampler~GreedySampler

表示贪婪采样器的类。

类型：generation/logits_sampler 的内部类

---

greedySampler.sample(logits) ⇒ <code>*</code>

对给定的 logits 张量采样最大概率。

类型：GreedySampler 的实例方法
返回值：* - 一个包含单个元组的数组，包含最大值的索引和一个无意义的分数（因为这是贪婪搜索）。

参数	类型
logits	张量

---

generation/logits_sampler~MultinomialSampler

表示多项式采样器的类。

类型：generation/logits_sampler 的内部类

---

multinomialSampler.sample(logits) ⇒ <code>*</code>

从 logits 中采样。

类型：MultinomialSampler 的实例方法

参数	类型
logits	张量

---

generation/logits_sampler~BeamSearchSampler

表示集束搜索采样器的类。

类型：generation/logits_sampler 的内部类

---

beamSearchSampler.sample(logits) ⇒ <code>*</code>

从 logits 中采样。

类型：BeamSearchSampler 的实例方法

参数	类型
logits	张量

---

< > 在 GitHub 上更新