BERT模型的输入

今天简单的介绍一下BERT的模型输入

上图显示的是BERT输入表示
总述:输入嵌入分别是token embeddings, segmentation embeddings 和position embeddings 的总和

BERT最主要的组成部分便是,词向量(token embeddings)、段向量(segment embeddings)、位置向量(position embeddings)

词向量:是模型中关于词最主要的信息
段向量:是因为BERT里面的下一句的预测任务,所以会有两句拼接起来,上句与下句,上句有上句段向量,下句则有下句段向量,也就是图中A与B。此外,句子末尾都有加[SEP]结尾符,两句拼接开头有[CLS]符
位置向量:是因为 Transformer 模型不能记住时序,所以人为加入表示位置的向量
之后这三个向量拼接起来的输入会喂入BERT模型,输出各个位置的表示向量

 

最近在做基于BERT的阅读理解,所以对BERT进行了进一步的学习,将自己平时的学习的过程进行了简单的记录

 

  • 8
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 3
    评论
在C++中构造BERT模型输入,可以使用第三方库(如TensorFlow C++ API或PyTorch C++ API)来加载预训练的BERT模型,并通过调用其相应的接口来构造输入。以下是一个基本的C++代码示例,展示了如何使用TensorFlow C++ API构造BERT模型输入: ```cpp #include <tensorflow/cc/client/client_session.h> #include <tensorflow/cc/ops/standard_ops.h> #include <tensorflow/core/framework/tensor.h> // 假设你已经有了一个函数 tokenize() 来进行分词 tensorflow::Tensor constructInput(const std::string& question, const std::string& text, int maxSequenceLength) { // 分词 std::vector<std::string> questionTokens = tokenize(question); std::vector<std::string> textTokens = tokenize(text); // 添加特殊标记和句子分隔符 std::vector<std::string> tokens = {"[CLS]"}; tokens.insert(tokens.end(), questionTokens.begin(), questionTokens.end()); tokens.push_back("[SEP]"); tokens.insert(tokens.end(), textTokens.begin(), textTokens.end()); tokens.push_back("[SEP]"); // 创建输入张量 tensorflow::Tensor input(tensorflow::DataType::DT_STRING, tensorflow::TensorShape({1, maxSequenceLength})); // 填充序列长度 int paddingLength = maxSequenceLength - tokens.size(); if (paddingLength > 0) { for (int i = 0; i < paddingLength; ++i) { tokens.push_back("[PAD]"); } } // 将分词后的词片段转化为字符串,并填充输入张量 for (int i = 0; i < maxSequenceLength; ++i) { input.flat<tensorflow::tstring>()(i) = tokens[i]; } return input; } int main() { std::string question = "aa"; std::string text = "bb"; int maxSequenceLength = 128; // 构造输入 tensorflow::Tensor input = constructInput(question, text, maxSequenceLength); // 加载预训练的BERT模型并进行推理... return 0; } ``` 请注意,上述代码示例中使用了TensorFlow C++ API,需要提前安装和配置TensorFlow C++库。此外,还需要根据具体的BERT模型和任务进行相应的调整和修改。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小杨算法屋

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值