如何从Keras使用其后端收集张量？

Question

我想在Keras中编译一个模型，其输入是2D numpy数组。 我需要的是将矢量放在这个二维数组的第n个位置，并将其用作其中一个图层的张量1D张量。 我该怎么做？

Answer 1

使用Lambda层应该这样做：

extracted_tensor = Lambda(lambda x: x[:,nth_index,:], output_shape=(1,dim_vector))(input) 
extracted_tensor = Flatten()(extracted_tensor) 

注意，在x张量（lambda函数），你采取批处理维度考虑，但你在output_shape参数没有。

我希望这可以帮助