在keras中使用常量向量连接输入。如何定义batch_size

Question

作为这个问题的后续行动：

Concatenate input with constant vector in keras

我正在尝试使用建议的解决方案：

constant=K.variable(np.ones((1,10, 5)))
constant = K.repeat_elements(constant,rep=batch_size,axis=0)

得到以下错误：

NameError: name 'batch_size' is not defined

我没有看到如何在keras模型中定义batch_size [未明确]，以便可以连接符号图层和常量图层，以便将它们用作输入图层。

Answer 1

获取动态批量大小：

batch_size = K.shape(your_tensor)[0]

但K.repeat_elements()不接受Tensor的{​​{1}}值。但是，您可以使用K.tile()生成相同的结果：

 

rep