白话transformer(二):Q K V矩阵

前面说了注意力机制的工作原理,本次来看看为了实现自注意力机制,Q K V矩阵是如何实现的。

B站视频

白话transformer(二)

1、语义相似性

我们在前面说了,embedding的作用就是相似的单词会被赋予相似的数字,那么我们是如何来计算语义的相似度的呢?
在这里插入图片描述
相似性其实是我们用来单词之间相似性的标准,我们希望有一种测量的方法使得可以用一些客观表示来形容相似度。

我们在这里介绍三种可以计算语义相似度的方法

1.1 点积

点积(dot product),又称内积(inner product),是两个向量的运算,结果是一个标量(一个单一的数值)。在数学、物理和计算机科学中,点积是一种基础且重要的运算。
对于两个向量

  • 16
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
Transformer中的QKV计算是通过将输入的特征向量分别映射为查询(Q)、键(K)和值(V)向量,然后通过计算注意力权重来获取最终的输出。下面是计算Transformer中的QKV的步骤: 1. 映射输入特征向量为查询(Q)、键(K)和值(V)向量[^1][^2]。 - 首先,通过线性变换将输入特征向量映射为查询向量Q、键向量K和值向量V。这个线性变换可以使用矩阵乘法和偏置项来实现。 - 例如,在自然语言处理任务中,输入特征向量可以是词嵌入向量,通过线性变换将其映射为查询向量Q、键向量K和值向量V。 2. 计算注意力权重[^1]。 - 使用查询向量Q和键向量K计算注意力权重。注意力权重表示了查询向量与键向量之间的相关性。 - 通过计算查询向量Q和键向量K的点积,然后除以一个缩放因子(如向量维度的平方根),再经过softmax函数得到注意力权重。 3. 使用注意力权重加权求和得到输出。 - 将注意力权重与值向量V相乘,然后对结果进行加权求和,得到最终的输出向量。 - 这个输出向量可以作为下一层的输入,或者作为模型的最终输出。 下面是一个示例代码,演示了如何计算Transformer中的QKV: ```python import torch import torch.nn as nn # 定义一个Transformer模型 class Transformer(nn.Module): def __init__(self, input_dim, output_dim): super(Transformer, self).__init__() self.linear_q = nn.Linear(input_dim, output_dim) self.linear_k = nn.Linear(input_dim, output_dim) self.linear_v = nn.Linear(input_dim, output_dim) def forward(self, x): q = self.linear_q(x) k = self.linear_k(x) v = self.linear_v(x) # 计算注意力权重 attention_weights = torch.softmax(torch.matmul(q, k.transpose(-2, -1)) / torch.sqrt(torch.tensor(q.size(-1))), dim=-1) # 使用注意力权重加权求和得到输出 output = torch.matmul(attention_weights, v) return output # 创建一个Transformer模型实例 input_dim = 512 output_dim = 256 transformer = Transformer(input_dim, output_dim) # 输入特征向量 x = torch.randn(10, input_dim) # 计算Transformer中的QKV output = transformer(x) print(output) ```

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

Andy_shenzl

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

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

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

打赏作者

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

抵扣说明:

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

余额充值