上传者: 38620734
|
上传时间: 2021-11-25 08:24:37
|
文件大小: 44KB
|
文件类型: -
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist=input_data.read_data_sets('mnist_data/',one_hot=True)
#注意这里用了one_hot表示,标签的形状是(batch_size,num_batches),类型是float,如果不用one_hot,那么标签的形状是(batch_size,),类型是int
num_classes=10
batch_size=64
hidden_dim1=32
hidden_dim2=64