generation/logits_sampler

• generation/logits_sampler
  • 静态
    • .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

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

类型：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	张量
索引	数字

---

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

表示 BeamSearchSampler 的类。

类型：generation/logits_sampler 的内部类

---

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

从 logits 中采样。

类型：BeamSearchSampler 的实例方法

参数量	类型
logits	张量

---

< > 在 GitHub 上更新