摘要: 用tf.keras写了自定义层,但在调用自定义层的时候总是报错,找了好久才发现问题所在,所以记下此问题。 问题代码 其中inputs的第一维为None,这里的代码为自定义的前向传播。我是想将得到的输出张量维度为1的维度删掉,因此调用了tf.squeeze方法,这时虽然没有报错但出现了问题。我分别打印 阅读全文
posted @ 2019-09-24 23:35 来一点音乐 阅读(500) 评论(0) 推荐(0) 编辑