95 Star 1.3K Fork 884

GVPMindSpore/mindformers

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
mindformers.models.glm2.ChatGLM2ForConditionalGeneration.rst 2.28 KB
一键复制 编辑 原始数据 按行查看 历史
wja 提交于 10个月前 . 修改原因:glm文档整改

mindformers.models.glm2.ChatGLM2ForConditionalGeneration

.. py:class:: mindformers.models.glm2.ChatGLM2ForConditionalGeneration(config: ChatGLM2Config, **kwargs)

    在线计算并提供执行ChatGLM2训练时的损失值和逻辑值。

    参数:
        - **config** (ChatGLM2Config) - ChatGLM2模型的配置。
        - **kwargs** (dict, 可选) - 一个可变数量的关键字参数,为待扩展的关键字参数预留。

    输入:
        - **input_ids** (Tensor) - 一个分词后的输入数据张量,它是32位整数类型,shape为: `(batch, seq_length)` 。默认值: ``None`` 。
        - **labels** (Tensor) - 一个分词后的标签数据张量,它是32位整数类型,shape为: `(batch, seq_length)` 。默认值: ``None`` 。
        - **input_position** (Tensor) - 当前位置,在推理使使用。默认值: ``None`` 。
        - **position_ids** (Tensor) - 保留参数,不使用。默认值: ``None`` 。
        - **attention_mask** (Tensor) - 保留参数,不使用。默认值: ``None`` 。
        - **input_embeds** (Tensor) - 保留参数,不使用。默认值: ``None`` 。
        - **init_reset** (bool) - shape为[1]的bool张量,用于清除增量推理中之前的键参数和值参数。默认值: ``None`` 。
        - **batch_valid_length** (Tensor) - 在增量推理中,用于上一步计算索引的张量。它是32位整数类型,shape为 `[batch_size]` 。默认值: ``None`` 。
        - **prefix_key_values** (Tensor) - 在正常的键值对之前添加的一组额外的键值对。这些前缀键值对可以用来捕获长期依赖关系或提供先验知识,从而帮助模型更好地理解和生成序列。默认值: ``None`` 。
        - **block_tables** (Tensor[int64]) - 存储每个序列的映射表。默认值: ``None`` 。
        - **slot_mapping** (Tensor[int32]) - 存储序列缓存的物理槽索引。默认值: ``None`` 。
        - **batch_index**  (Tensor) - 保留参数,不使用。默认值: ``None`` 。
        - **zactivate_len** (Tensor) - 保留参数,不使用。默认值: ``None`` 。

    输出:
        outputs(Tensor),包括在线损失值或者逻辑值、预测文本序列、输入掩码。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindformers.git
git@gitee.com:mindspore/mindformers.git
mindspore
mindformers
mindformers
v1.3.0

搜索帮助