Tensorflow 合并通道及加载子模型的方法

上传:Lainah 浏览: 10 推荐: 0 文件:PDF 大小:60.37KB 上传时间:2020-12-31 19:02:12 版权申诉
最近在使用Tensorflow 实现DNN网络时,遇到一些问题。目前网上关于Tensorflow的资料还比较少,现把问题和解决方法写出来,仅供参考。 (1)将两个子模型的输出合并到一个通道,例如同时连接到一个全连接层如图 合并方法为 tf.concat()函数。此函数需要两个个参数 concat(0或1,[合并节点1,合并节点2] )。0 或 1 代表节点合并的方式:0 代表合并后列相同,行增加;1 代表合并后行相同,列增加。 上图所示合并方法为: X_20 = tf.concat(1, [X_top, X_down]); (2)加载预训练好模型的部分参数。例如训练完成了一个五层网络,现在需
上传资源
用户评论