该tf.layers模块包含创建上述三种类型中的每一种的方法:
conv2d()。构造二维卷积层。获取过滤器的数量,过滤内核大小,填充和激活功能作为参数。
max_pooling2d()。...如果要为宽度和高度设置不同的步幅值,可以指定一个元组或列表(例如,stride=[3,6])。...为了计算交叉熵,首先我们需要转换labels 为相应的 单热编码:
[[0, 1, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 1],...这depth是10因为我们有10个可能的目标类,每个数字一个。
接下来,我们onehot_labels从我们的逻辑层计算预测的交叉熵和softmax。...(
mode=mode, predictions=predictions, loss=loss, train_op=train_op)
训练和评估CNN MNIST分类器
我们编码了我们的