Tensorflow 基于minst手写数字数据集合建立RNN分类模型

rnn网络架构

读取数据

import tensorflow as tf
import input_data
import numpy as np
import matplotlib.pyplot as plt
print ("Packages imported")
import warnings

mnist = input_data.read_data_sets("data/", one_hot=True)
trainimgs, trainlabels, testimgs, testlabels = mnist.train.images, mnist.train.labels, mnist.test.images, mnist.test.labels 
ntrain, ntest, dim, nclasses  = trainimgs.shape[0], testimgs.shape[0], trainimgs.shape[1], trainlabels.shape[1]
print ("MNIST loaded")

 

初始化参数

diminput  = 28
dimhidden = 128
dimoutput = nclasses
nsteps    = 28
weights = {
    '