使用Lambda解决_inbound_nodes错误

Keras出现了下面的错误:

AttributeError: ‘NoneType’ object has no attribute ‘_inbound_nodes’
原因是使用了Keras backend的reshape操作:

x = K.reshape(x, (num_pictures, 32, 32, 512))
但是Keras backend并不是一个Layer,于是出现了上面的错误.解决的方法是使用Lambda,Lambda用于定义一个Layer,其中没有需要学习的变量,只是对Tensor进行一些操作.先定义一个reshape的函数:

def reshape_tensor(x, shape):
return K.reshape(x, shape);
然后再调用这个函数:

x = Lambda(reshape_tensor, arguments={‘shape’: (num_pictures, 32, 32, 512)})(x)
这样就不会出错了.

原文:https://www.cnblogs.com/mstk/archive/2019/06/12/11008901.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值