利用自己搭建的神经网络实现输出我想要的结果，但是在训练的过程中可能在输入的维度上出现了问题，但是不知道问题出在了那里。

Question

这第一张图片是我的报错信息，下面这张图是我搭建的神经网络，我期望他的输入是一个4*201的矩阵，不知道如图定义的网络对不对。

下面这张图片是我的我输入的数据的集合，不知道是不是在建立训练数据集的时候数据集的哪个维数出现了错误。