tensorflow rnn 最简单实现代码

xiaoxiao2021-02-28  108

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

#!/usr/bin/env python # -*- coding: utf-8 -*- import tensorflow as tf from tensorflow.contrib import rnn import numpy as np x=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/
转载请注明原文地址: https://www.6miu.com/read-28225.html

最新回复(0)