自从安装了tensorflow 2.0 ,我就没有一天不头疼的 ,还因为这个天天发牢骚,差点放弃深度学习 hahah
然后今天静下心来整理了一系列的问题,其实我觉得解决的根本就是弄懂她
我们看到在1.x的版本中,这个contrib是已经不能用的了,所以在2.x中 如果你遇见了tf.contrib可能你就要注意了,我先说说我遇见的两处:
# 一维化上一层的输出
P = tf.contrib.layers.flatten(P2)
这一句当时就报错了,我遇见这种问题的第一反应百度,也确实有人了遇见这个问题,所以很轻松解决了,但是我觉得我们以后遇见这种问题,可以直接去tensorflow2.x的官网API文档中,找对应的升级后的代码:
我们可以看到这里有好几个flatten,但是当我们点进去的时候会发现具体的功能是不一样的,我们现在想做的是一维化:
点进去看这个tf.nest.flatten我们发现它是返回给定嵌套结构中的平面列表,所以不是我们要的函数。
这里是正确的2.x版本中的,我们从它函数的解释中就能明白这个是我们