tensorflow rnn 最简单实现代码

来源:互联网 发布:淘宝账号的心怎么涨 编辑:程序博客网 时间:2024/05/17 19:57

因为现在的 example 都比较复杂涉及东西比较多,抽出来一个极简版本。


#!/usr/bin/env python# -*- coding: utf-8 -*-import tensorflow as tffrom tensorflow.contrib import rnnimport numpy as npx=tf.placeholder(dtype=tf.float64,shape=[10,10,10],name="x")train_x = np.ones(shape=[10, 10, 10], dtype=float)cell=tf.nn.rnn_cell.BasicLSTMCell(10)unstack_x = tf.unstack(x, 10, 1)lstm_cell = rnn.BasicLSTMCell(10, forget_bias=1.0)outputs, states = rnn.static_rnn(lstm_cell, unstack_x, dtype=tf.float64)with tf.Session() as sess:sess.run(tf.global_variables_initializer())outputs_array=(sess.run(outputs,feed_dict={x:train_x}))



更多教程:http://www.tensorflownews.com/

原创粉丝点击