比如一张彩色照片由宽 28,高 28,3 色道的元素组成,那么该三维照片数据的形状 (完全可知) 可写成
[28, 28, 3]
将照片当做神经网络的输入时,有时候我们不清楚有多少张照片,因此整个四维照片数据集的形状...但本节重点只看张量运算,因此我们只关注训练集,发现它的形状是 (60000, 28, 28),其中 60000 代表图片个数。...注意 reshape 函数第二个参数用到了 -1,这是我们偷懒不想计算 28×28。本来重塑后的形状是 (60000, 784),转置之后 X 形状是 (784, 60000)。...根据 X 的形状为 (784, 60000),可先推出 W 的形状应该为 (?...,784);又根据输出 Y 包含 10 个数字的概率值,因此 b 的形状为 (10, 1),则完全推出 W 的形状 (10, 784)。