[发明专利]运算装置及相关产品

权利要求书

1.一种运算装置，其特征在于，所述运算装置包括：主指令处理单元、主内存单元以及主功能单元，

其中，所述主指令处理单元用于在接收到输入指令后，根据输入指令向所述主内存单元以及所述主功能单元发送第一控制信号；

所述主内存单元用于根据所述第一控制信号向所述主功能单元发送输入数据，所述输入数据以张量形式表示；

所述主功能单元用于根据所述第一控制信号将输入数据的winograd正变换拆解为求和运算，并进行计算得到所述输入数据的winograd正变换结果。

2.根据权利要求1所述的运算装置，其特征在于，

所述主功能单元用于根据所述第一控制信号将所述输入数据拆解为多个第一子张量，对所述多个第一子张量进行winograd正变换并求和得到所述输入数据的winograd正变换结果。

3.根据权利要求2所述的运算装置，其特征在于，所述多个第一子张量的个数与所述输入数据的不为0的元素的个数相同，所述多个第一子张量中的每个第一子张量中有一个元素与所述输入数据中的对应位置的元素相同、其他元素均为0。

4.根据权利要求3所述的运算装置，其特征在于，对所述多个第一子张量进行winograd正变换并求和得到所述输入数据的winograd正变换结果，包括：

获取第一子张量对应的第一元子张量的winograd正变换结果；其中，第一子张量对应的第一元子张量为：在第一元子张量中第一位置的元素的值为1，其中，第一位置在第一元子张量中所处的位置与第一子张量中的非0元素所处的位置相同；

将第一子张量中不为0的元素值作为系数乘以对应的第一元子张量的winograd正变换结果，得到第一子张量的winograd正变换结果；

将多个第一子张量的winograd正变换结果相加得到所述输入数据的winograd正变换结果。

5.根据权利要求4所述的运算装置，其特征在于，第一子张量对应的第一元子张量的winograd正变换结果是通过以下过程预先得到的：

对于每一个第一子张量，将该第一子张量对应的第一元子张量左边乘以正变换左乘矩阵、右边乘以正变换右乘矩阵得到第一元子张量的winograd正变换结果。

6.根据权利要求1所述的运算装置，其特征在于，所述主功能单元包括缓存模块，所述主功能单元将所述输入数据的winograd正变换结果存储到所述缓存模块中。

7.根据权利要求1-6任意一项所述的运算装置，其特征在于，所述输入数据为输入神经元或者权值。

8.一种运算装置，其特征在于，所述运算装置包括：主指令处理单元、主功能单元、从指令处理单元、以及从功能单元，

所述从指令处理单元用于接收主指令处理单元发送的第二控制信号，并将所述第二控制信号发送给所述从功能单元；所述从功能单元用于接收主功能单元发送的输入数据的winograd正变换结果，所述输入数据的winograd正变换结果包括输入神经元的winograd正变换结果；

所述功能单元用于根据所述第二控制信号对输入神经元的winograd正变换结果和权值的winograd正变换结果进行对位乘得到对位乘结果，将所述对位乘结果的winograd逆变换拆解为求和运算，并进行计算得到所述输入数据的winograd卷积结果。

9.根据权利要求8所述的运算装置，其特征在于，所述从功能单元用于将所述对位乘结果拆解为多个第二子张量，对所述多个第二子张量进行winograd逆变换并求和得到所述输入数据的winograd卷积结果。

10.根据权利要求9所述的运算装置，其特征在于，所述多个第二子张量的个数与对位乘结果的非0元素的个数相同，所述多个第二子张量中的每个第二子张量中有一个元素与对位乘结果中的对应位置的元素相同、其他元素均为0。