tensorflow - 如何在 Keras 模型定义中索引张量列表

Question

假设我们对 keras 模型有如下定义

def build_model(ls):
    x = Input(shape=(1,), name='input_index')
    gs = [K.constant(x) for x in ls]

    idx = Lambda(lambda x: K.cast(x, 'int32'))(x)
    y = Lambda(lambda x: gs[x[0, 0]])(idx)

    model = Model(inputs=x, outputs=y)
    return model

在这里，我们传递一个列表ndarray:ls来初始化模型。在模型定义中，gs转换ls为常量张量列表。输入x给出了索引i，模型应该已经返回了相应的张量gs[i]。但是，上面给出的模型是正确的。

有什么办法可以在 keras 中做这样的事情吗？

标签： tensorflowkeras