Keras模型Tensorflow Lite转换输入形状

Question

我正在尝试将简单的Keras模型冻结图转换为tensorflowlite，但是我不确定输入的形状是什么。

    toco 
--input_file='my_model.pb'
--input_format=TENSORFLOW_GRAPHDEF \
--output_format=TFLITE \
--output_file=/tmp/my_model.tflite \
--inference_type=FLOAT \
--input_type=FLOAT \
--input_arrays=input_tensor \
--output_arrays=output_tensor \
--input_shapes=0,4,2851

我的模特是：

# create model
model = Sequential()
model.add(Dense(50, activation="tanh", input_dim=4, kernel_initializer="random_uniform", name="input_tensor"))
model.add(Dense(50, activation="tanh", kernel_initializer="random_uniform"))
model.add(Dense(1, activation="linear", kernel_initializer='random_uniform', name="output_tensor"))