TypeError: __call__() takes 2 positional arguments but 3 were given

在keras利用lambda函数添加层时,lambda需要输入两个变量时出现的bug

这个bug困扰了我两天,一直困惑为什么会出现这种问题,一直很费解~~~
代码如下:

import keras
import tensorflow as tf

get_lambda = lambda a,b: tf.matmul(a,b)
get_lambda(tf.zeros((1,2)),tf.zeros((2,1)))
get_lambda_layer = keras.layers.Lambda(get_lambda)
get_lambda_layer(tf.zeros((1,2)),tf.zeros((2,1)))

报错:

TypeError: __call__() takes 2 positional arguments but 3 were given

很懵逼,明明我输入了两个变量。。。。
好吧,那我把他们括起来。

get_lambda_layer([tf.zeros((1,2)),tf.zeros((2,1))])

还会报错:

TypeError: <lambda>() missing 1 required positional argument: 'b'

找不到第二个变量???
原来,它认为 a = [tf.zeros((1,2)),tf.zeros((2,1))]
好吧。。。。

解决方案:

注意这里将lambda添加层时,只允许有一个变量,而且是tensor 或 tensor 列表
这样,代码修改为:

get_lambda2 = lambda x: tf.matmul(x[0],x[1])
get_lambda_layer2 = keras.layers.Lambda(get_lambda2)
get_lambda_layer2([tf.zeros((1,2)),tf.zeros((2,1))])

运行:

<tf.Tensor 'lambda_9/MatMul:0' shape=(1, 1) dtype=float32>

OK! 问题完美解决!!! 这个大坑啊~~~~

  • 6
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值