矩阵乘法脉动阵列的C++模拟

自从谷歌的TPU问世以后，被人们遗忘很久的脉动阵列又再次火了一把。矩阵乘法就可以用脉动阵列进行计算，而脉动阵列这种数据流又特别适合用硬件进行实现。下面是用脉动阵列进行矩阵乘法的示意图。可以看到，A的每一行不是同时进入脉动阵列的，而B的每一列也不是同时进入脉动阵列的，相邻行或列进入脉动阵列的时间恰好相差一个时钟周期。 以下是用c++模拟脉动阵列的代码：

#include 
#include 
#include 
#include 
#define N 256
using namespace std;

typedef struct PE{
    int weight;
    int neuron;
    int psum;
};

class Systolic{
public:
    PE S[N][N];
public:
    void Init(){
        for(int i=0;i