循环计算层的tensorflow描述

循环计算层的tensorflow描述,第1张

TF描述循环计算层:

tf.keras.layers.SimpleRNN(循环核中记忆体的个数,activation=‘激活函数’,return_sequences=是否每个时刻输出到ht到下一层)

activation=‘激活函数’:表示使用什么激活函数计算ht,如果不写,默认tanh
return_sequences=True:表示各时间步输出ht到下一层
return_sequences=False:表示仅在最后时间步输出ht(默认)
return_sequences=True:

return_sequences=False:

一般最后一层的循环核用False,中间层的循环核用True:表示每个时间步都把参数输出到下一层,仅在最后时间步输出ht
如下图:

如tf.keras.layers.SimpleRNN(3,return_sequences=True)

要求送入RNN的数据是三维的,即x_train是三维的:
[送入样本数,循环核时间展开步数,每个时间步输入特征个数]

如上图的第一个图:以关送入RNN2组数据,每组数据经过1个时间步即可输出结果,每个时间步送入3个数值。


所以是[2,1,3]

欢迎分享,转载请注明来源:内存溢出

原文地址: http://www.outofmemory.cn/langs/578741.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-11
下一篇 2022-04-11

发表评论

登录后才能评论

评论列表(0条)

保存