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: () 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))])

运行:


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

你可能感兴趣的:(keras)