[发明专利]一种神经网络的全连接层计算方法和装置有效
申请号: | 202010725384.9 | 申请日: | 2020-07-24 |
公开(公告)号: | CN111860838B | 公开(公告)日: | 2022-12-20 |
发明(设计)人: | 宋小妹 | 申请(专利权)人: | 苏州浪潮智能科技有限公司 |
主分类号: | G06N3/08 | 分类号: | G06N3/08;G06N3/04 |
代理公司: | 北京连和连知识产权代理有限公司 11278 | 代理人: | 杨帆;宋薇薇 |
地址: | 215100 江苏省苏州市吴*** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开了一种神经网络的全连接层计算方法和装置,方法包括:接收格式为8位整型变量的输入数据,并判定输入数据的输入维度、输出维度、和单次样本是否满足张量核的调用接口的形状要求;响应于满足形状要求而基于8位整型变量的输入数据、8位整型变量的权重、和32位整型变量的偏置来定义计算规则;基于计算规则调度输入数据、权重、和偏置,以通过调用接口调用张量核的计算单元进行矩阵乘法运算,获得二维张量;将二维张量作为全连接层的计算结果输出到下一层。本发明能够在dense层针对int8输入调用TensorCore计算单元执行卷积,大幅度提升计算性能。 | ||
搜索关键词: | 一种 神经网络 连接 计算方法 装置 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于苏州浪潮智能科技有限公司,未经苏州浪潮智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202010725384.9/,转载请声明来源钻瓜专利网。