首页 > 解决方案 > 如何处理 Keras 层的重复输入?

问题描述

我有一个 Keras 模型,它有两个输入层。

  1. 一条形状的推文(20,300)
  2. 其他五条形状的推文(5,20,300)。但是,此输入对于所有训练示例都是相同的。

换句话说,对于每个训练步骤,都会有不同的推文(第一个输入)和相同的五个推文(第二个输入)。我的第二个输入的形状(5,20,300)非常大,可以重复num_samples多次,然后用作 Keras 模型的输入层。我需要一种方法来使第二个输入在 keras 模型中使用但不重复num_samples

有没有办法处理这种类型的输入?

标签: pythontensorflowkerasdeep-learningkeras-layer

解决方案


使用该常量输入创建一个张量:

fixed_tweets = keras.backend.constant(the_tweets_as_numpy)

使用常规输入和tensor输入:

input1 = Input((20,300))
input2 = Input(tensor=fixed_tweets)

去玩吧!!

您可能需要自定义层来处理input1(any) 和input2(5) 的批量大小之间的差异。


推荐阅读