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) ⇒ *

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

类型：LogitsSampler 的实例方法

参数	类型
logits	Tensor

---

logitsSampler.sample(logits) ⇒ *

用于采样 logits 的抽象方法。

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

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

参数	类型
logits	Tensor

---

logitsSampler.getLogits(logits, index) ⇒ Float32Array

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

类型：LogitsSampler 的实例方法

参数	类型
logits	Tensor
index	number

---

logitsSampler.randomSelect(probabilities) ⇒ number

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

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

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

---

LogitsSampler.getSampler(generation_config) ⇒ LogitsSampler

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

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

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

---

generation/logits_sampler~GreedySampler

表示 Greedy Sampler 的类。

类型：generation/logits_sampler 的内部类

---

greedySampler.sample(logits) ⇒ *

采样给定 logits tensor 的最大概率。

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

参数	类型
logits	Tensor

---

generation/logits_sampler~MultinomialSampler

表示 MultinomialSampler 的类。

类型：generation/logits_sampler 的内部类

---

multinomialSampler.sample(logits) ⇒ *

从 logits 中采样。

类型：MultinomialSampler 的实例方法

参数	类型
logits	Tensor

---

generation/logits_sampler~BeamSearchSampler

表示 BeamSearchSampler 的类。

类型：generation/logits_sampler 的内部类

---

beamSearchSampler.sample(logits) ⇒ *

从 logits 中采样。

类型：BeamSearchSampler 的实例方法

参数	类型
logits	Tensor

---

< > 在 GitHub 上更新