输入为一个向量784， 也就是图片的像素长x宽。
784 = 28 * 28
keepdim表示保持维度。以0的方向相加，表示去掉row的维度，也就是结果为1row；以1的方向相加，表示去掉col的维度，也就是结果为1col。
X.reshape((-1, W.shape[0])), 维度改变为bathSize * 784， bathSize = 256
https://www.bilibili.com/video/BV1K64y1Q7wu
