我试图在keras lambda层中使用tensorflow唯一函数(https://www.tensorflow.org/api_docs/python/tf/unique)。代码如下:
def unique_idx(x):
output = tf.unique(x)
return output[1]
then
inp1 = Input(batch_shape(None, 1))
idx = Lambda(unique_idx)(inp1)
model = Model(inputs=inp1, outputs=idx)
Run Code Online (Sandbox Code Playgroud)
现在使用时**model.compile(optimizer='Adam', loss='mean_squared_error')**
出现错误:
ValueError:张量转换请求的dtype float32的张量为dtype int32:'Tensor(“ lambda_9_sample_weights_1:0”,shape =(?,),dtype = float32)'
有人知道这里的错误是什么或使用张量流函数的其他方式吗?