【技术实现步骤摘要】
运算方法、装置及相关产品
本公开涉及信息处理
,尤其涉及一种运算方法、装置及相关产品。
技术介绍
在人工智能
,神经网络算法是最近非常流行的一种机器学习算法,在各种领域中都取得了非常好的效果,比如图像识别,语音识别,自然语言处理等。随着神经网络算法的发展,算法的复杂度也越来越高,为了提高识别度,模型的规模也在逐渐增大。
技术实现思路
根据本公开的第一方面,提供了一种矩阵的运算方法,所述方法包括:根据接收的待运算矩阵,获取对应的待运算张量,所述待运算张量包括第一张量和第二张量;根据所述第一张量和所述第二张量的形状,确定神经网络模型的输入数据;将所述输入数据通过所述神经网络模型,根据所述神经网络模型的输出结果,得到所述待运算矩阵的运算结果。在一种可能的实现方式中,根据接收的待运算矩阵,获取对应的第一张量和第二张量,所述待运算张量包括第一张量和第二张量,包括:接收第一矩阵和第二矩阵;根据所述第一矩阵和所述第二矩阵,分别得到对应的第一张量和第二张量。在一种可能的实现方式中,根据所述第一矩阵和所述第二矩阵,分别得到对应的第一张量和第二张量,包括:对所述第一矩阵和所述第二矩阵分别进行相同的维度扩展,得到对应的第一张量和第二张量。在一种可能的实现方式中,根据所述第一张量和所述第二张量的形状,确定神经网络模型的输入数据,包括:将所述第一张量作为神经网络模型的初始输入张量,所述第二张量作为神经网络模型的初始过滤器张量,得到所 ...
【技术保护点】
1.一种矩阵的运算方法,其特征在于,所述方法包括:/n根据接收的待运算矩阵,获取对应的待运算张量,所述待运算张量包括第一张量和第二张量;/n根据所述第一张量和所述第二张量的形状,确定神经网络模型的输入数据;/n将所述输入数据通过所述神经网络模型,根据所述神经网络模型的输出结果,得到所述待运算矩阵的运算结果。/n
【技术特征摘要】
1.一种矩阵的运算方法,其特征在于,所述方法包括:
根据接收的待运算矩阵,获取对应的待运算张量,所述待运算张量包括第一张量和第二张量;
根据所述第一张量和所述第二张量的形状,确定神经网络模型的输入数据;
将所述输入数据通过所述神经网络模型,根据所述神经网络模型的输出结果,得到所述待运算矩阵的运算结果。
2.根据权利要求1所述的矩阵的运算方法,其特征在于,根据接收的待运算矩阵,获取对应的第一张量和第二张量,所述待运算张量包括第一张量和第二张量,包括:
接收第一矩阵和第二矩阵;
根据所述第一矩阵和所述第二矩阵,分别得到对应的第一张量和第二张量。
3.根据权利要求2所述的矩阵的运算方法,其特征在于,根据所述第一矩阵和所述第二矩阵,分别得到对应的第一张量和第二张量,包括:
对所述第一矩阵和所述第二矩阵分别进行相同的维度扩展,得到对应的第一张量和第二张量。
4.根据权利要求1所述的矩阵的运算方法,其特征在于,根据所述第一张量和所述第二张量的形状,确定神经网络模型的输入数据,包括:
将所述第一张量作为神经网络模型的初始输入张量,所述第二张量作为神经网络模型的初始过滤器张量,得到所述神经网络模型的初始输入数据;
根据所述初始输入张量和所述初始过滤器张量之间的比较结果,确定所述神经网络模型的输入数据。
5.根据权利要求4所述的矩阵的运算方法,其特征在于,根据所述初始输入张量和所述初始过滤器张量之间的比较结果,确定所述神经网络模型的输入数据,包括:
读取所述初始输入张量的批处理数目;
读取所述初始过滤器张量的输出通道数目;
根据所述批处理数目和所述输出通道数目的大小关系,确定所述神经网络模型的输入数据。
6.根据权利要求5所述的矩阵的运算方法,其特征在于,根据所述批处理数目和所述输出通道数目的大小关系,确定所述神经网络模型的输入数据,包括:
在所述输出通道数目与所述批处理数目之差不小于阈值时,对所述初始输入数据执行调换操作,得到所述神经网络模型的输入数据;
在所述输出通道数目与所述批处理数目之差小于阈值时,将所述初始输入数据作为所述神经网络模型的输入数据。
7.根据权利要求6所述的矩阵的运算方法,其特征在于,对所述初始输入数据执行调换操作,得到所述神经网络模型的输入数据,包括:
将所述初始输入张量作为所述神经网络模型的过滤器张量,将所述初始过滤器张量作为所述神经网络模型的输入张量,得到所述神经网络模型的输入数据。
8.根据权利要求1所述的矩阵的运算方法,其特征在于,将所述输入数据通过所述神经网络模型,根据所述神经网络模型的输出结果,得到所述待运算矩阵的运算结果,包括:
将所述输入数据通过所述神经网络模型,得到所述神经网络模型的输出结果;
根据矩阵运算的维度要求,提取所述输出结果中相应维度的数据,得到所述待运算矩阵的运算结果。
9...
【专利技术属性】
技术研发人员:不公告发明人,
申请(专利权)人:上海寒武纪信息科技有限公司,
类型:发明
国别省市:上海;31
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。