小编Nic*_*wes的帖子

在keras层中包装张量流功能

我试图在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)'

有人知道这里的错误是什么或使用张量流函数的其他方式吗?

python python-3.x keras

2
推荐指数
1
解决办法
3127
查看次数

标签 统计

keras ×1

python ×1

python-3.x ×1