在 Keras 中创造恒定的价值

Question

我正在尝试在 keras 模型中创建一个常量变量。到目前为止我所做的是将它作为输入传递。但它始终是一个常数，所以我希望它是一个常数。（每个示例的输入是[1,2,3...50] => 所以我使用np.tile(np.array(range(50)),(len(X_input))) 为每个示例重现它）

所以现在我有：

constant_input = Input(shape=(50,), dtype='int32', name="constant_input")

给出张量：Tensor("constant_input", shape(?,50), dtype=int32)

现在尝试将其作为常量：

np_constant = np.array(list(range(50))).reshape(1, 50)
tf_constant = K.constant(np_constant)
tensor_constant = Input(tensor=tf_constant, shape=(50,), dtype='int32', name="constant_input")

给出张量：Tensor("constant_input", shape(50,1),dtype=float32)

但我想要的是每批要缩放的常数，也就是说张量的形状应该是(?, 50)，和Input的使用方式一样。

有可能吗？

Answer 1

你不能有一个可变大小的常量。常数始终具有相同的值。您可以做的是拥有(1, 50) 常量，然后使用K.tile 在TensorFlow 中平铺它。最好使用np.arange 而不是np.array(list(range(50))。比如：

from keras.layers.core import Lambda
import keras.backend as K

def operateWithConstant(input_batch):
    tf_constant = K.constant(np.arange(50).reshape((1, 50)))
    batch_size = K.shape(input_batch)[0]
    tiled_constant = K.tile(tf_constant, (batch_size, 1))
    # Do some operation with tiled_constant and input_batch
    result = ...
    return result

input_batch = Input(...)
input_operated = Lambda(operateWithConstant)(input_batch)
# continue...