keras自定义层Lambda 输出维度不对

天下第一小白 天下第一小白 墨衍会员 · AI 创作全网分发 墨衍智能分发 通过墨衍一键同步至各平台 可同步平台 CSDN微信公众号微博知乎掘金百家号博客园抖音小红书 你的文章也可以这样分发 墨衍会员 ¥399起/年,写一次发全网 我也要 + 加入社区

def reduce_sums(self, vec):
user_out = vec[0]
item_out = vec[1]
res = tf.nn.sigmoid(tf.reduce_sum(tf.multiply(user_out, item_out), axis=1, keep_dims=False))
return res
prediction = Lambda(self.reduce_sums)([user_latent, item_latent])



ValueError: Error when checking target: expected lambda_2 to have 1 dimensions, but got array with shape (5167, 1)

lambda_1 (Lambda) (None,) 0
...全文

相关推荐

评论 1
收藏 已收藏
分享
返回顶部