首页> 中国专利> 注意力机制的激活函数加速方法、装置和注意力机制电路

注意力机制的激活函数加速方法、装置和注意力机制电路

摘要

本发明实施方式提出一种注意力机制的激活函数加速方法、装置和注意力机制电路。方法包括:输入第一向量和包含多个第二向量的第一向量组;基于注意力机制,确定所述第一向量对所述第一向量组中的每个第二向量的注意力分数;对所述注意力分数施加激活函数,所述激活函数适配于将所述注意力分数中的负值转变为零且正值保持为正数;基于预定的缩放因子,对施加所述激活函数的注意力分数进行缩放;输出所述缩放后的注意力分数。本发明实施方式不再采用Softmax计算注意力机制的激活函数,规避了以自然底数e为底数的、注意力分数为指数的幂运算,提高了注意力机制的运算速度,尤其适用于以硬件方式实现注意力机制。

著录项

  • 公开/公告号CN114528989A

    专利类型发明专利

  • 公开/公告日2022-05-24

    原文格式PDF

  • 申请/专利权人 深圳比特微电子科技有限公司;

    申请/专利号CN202210432779.9

  • 申请日2022-04-24

  • 分类号G06N3/063;G06N3/06;G06N3/04;G06F17/16;

  • 代理机构北京德琦知识产权代理有限公司;

  • 代理人张驰;宋志强

  • 地址 518000 广东省深圳市南山区高新南六道航盛科技大厦801

  • 入库时间 2023-06-19 15:24:30

法律信息

  • 法律状态公告日

    法律状态信息

    法律状态

  • 2022-05-24

    公开

    发明专利申请公布

相似文献

  • 专利
  • 中文文献
  • 外文文献
获取专利

客服邮箱:kefu@zhangqiaokeyan.com

京公网安备:11010802029741号 ICP备案号:京ICP备15016152号-6 六维联合信息科技 (北京) 有限公司©版权所有
  • 客服微信

  • 服务号