输入一个手写数字灰度图,输出预测
一个 [28, 28] 的灰度图,上面的数字是2
[0., 0., 1., 0., 0., 0., 0., 0., 0., 0.]
输入是 float32,范围在0到1之间 输出是 float32 的 one-hot 编码,必须是 0 或者 1