python - 是否可以使用 tf.keras.models.clone_model 来改变数据输入的形状？

Question

我想使用函数 API 创建的 tensorflow/keras 模型的方法clone_model并tf.keras更改其输入形状。因此，我尝试使用参数input_tensor来改变形状。但是，它似乎没有使用提供input_tensors的名称和形状与原始模型保持相同。论证的input_tensors目的是什么？

编码：

import tensorflow as tf
from tensorflow.keras import layers

inputs_small = layers.Input((64, 64, 3), name="small")

outputs = layers.Conv2D(32, 1)(inputs_small)
model_small = tf.keras.models.Model(inputs=inputs_small, outputs=outputs)


inputs_large = layers.Input((128, 128, 3), name="large")
model_large = tf.keras.models.clone_model(model_small, input_tensors=inputs_large)

model_large.summary()

结果是：

_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
small (InputLayer)           [(None, 64, 64, 3)]       0         
_________________________________________________________________
conv2d (Conv2D)              (None, 64, 64, 32)        128       
=================================================================

但我喜欢：

_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
large (InputLayer)           [(None, 128, 128, 3)]     0         
_________________________________________________________________
conv2d (Conv2D)              (None, 128, 128, 32)      128       
=================================================================

我使用 TensorFlow 2.4.1。我简化了我的问题。在我的代码中，我还使用了clone_function替换clone_model图层的参数。

标签： pythontensorflowkerastf.keras

Answer 1

我进一步调查并发现了一个 Keras 错误： https ://github.com/keras-team/keras/issues/14937