基础的RNN

2021/5/7 18:27:29

本文主要是介绍基础的RNN,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!


import tensorflow as tf
import numpy as np
n_input=3
n_neurons=5
x0=tf.placeholder(tf.float32,[None,n_input])
x1=tf.placeholder(tf.float32,[None,n_input])

wx=tf.Variable(tf.random_normal(shape=[n_input,n_neurons],dtype=tf.float32))
wy=tf.Variable(tf.random_normal(shape=[n_neurons,n_neurons],dtype=tf.float32))
b=tf.Variable(tf.zeros([1,n_neurons],dtype=tf.float32))

y0=tf.tanh(tf.matmul(x0,wx)+b)
y1=tf.tanh(tf.matmul(y0,wy)+tf.matmul(x1,wx)+b)

init=tf.global_variables_initializer()
x0_batch=np.array([[0,1,2],[3,4,5],[6,7,8],[9,0,1]])
x1_batch=np.array([[9,8,7],[0,0,0],[6,5,4],[3,2,1]])
with tf.Session() as sess:
   init.run()
   yo_val,y2_val=sess.run([y0,y1],feed_dict={x0:x0_batch,x1:x1_batch})
   print(yo_val)
   print('---------')
   print(y2_val)


这篇关于基础的RNN的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程