[发明专利]多运算神经网络单元

摘要

神经网络单元包括一神经处理单元阵列。N个神经处理单元的第一与第二多任务器分别选择性地集体作为第一与第二N文字旋转器。第一与第二存储器分别装载N个权重/数据文字列并将一列N个权重/数据文字提供至相对应的神经处理单元。神经处理单元选择性地：对来自第一存储器的多列权重文字与利用第二N文字旋转器接收自第二存储器的一列数据文字执行乘法累加运算；对利用第一N文字旋转器接收自第一存储器的多列权重文字与来自第二存储器的多列数据文字执行卷积运算；以及对利用第一N文字旋转器接收自第一存储器的多列权重文字执行共源运算。

主权项

1.一种神经网络单元，其特征在于，包括：一个由N个神经处理单元构成的阵列，各该神经处理单元包括：一算术单元与一累加器；以及第一与第二多任务缓存器，分别具有第一与第二输出，由该算术单元与一相邻神经处理单元的相对应第一与第二多任务器接收，该N个神经处理单元的该些第一多任务器与该些第二多任务器分别选择性地集体作为第一N个文字旋转器与第二N个文字旋转器进行运作；一第一存储器，以装载N个权重文字的列，并将一列该N个权重文字提供至该N个神经处理单元中的相对应的该些神经处理单元；以及一第二存储器，以装载N个数据文字的列，并将一列该N个数据文字提供至该N个神经处理单元中的相对应的该些神经处理单元；其中，该神经网络单元可经程序化使该神经处理单元阵列选择性地：对接收自该第一存储器的多列的N个权重文字与利用该第二N个文字旋转器接收自该第二存储器的一列N个数据文字，执行乘法累加运算；对利用该第一N个文字旋转器接收自该第一存储器的多列的N个权重文字与接收自该第二存储器的多列的N个数据文字，执行卷积运算，该多列权重文字为一数据矩阵，该多列数据文字为一卷积核的元素；以及对利用该第一N个文字旋转器接收自该第一存储器的多列的N个权重文字，执行共源运算。