无法将输入数组从形状 (150,150,3) 广播到形状 (150,150,3,3)答案

【问题标题】：could not broadcast input array from shape (150,150,3) into shape (150,150,3,3)无法将输入数组从形状 (150,150,3) 广播到形状 (150,150,3,3)
【发布时间】：2020-11-23 00:50:50
【问题描述】：

我尝试训练模型

EPOCHS = 5
history = model.fit(train_data, steps_per_epoch=len(train_generator), epochs=EPOCHS, validation_data=val_data, validation_steps=len(val_generator))

但我收到此错误：

值错误：无法从形状 (150,150,3) 广播输入数组成形 (150,150,3,3)。

谁能帮帮我？

【问题讨论】：

标签： arrays tensorflow input model

【解决方案1】：

如果您将形状 (150,150,3) 的输入更改为 (150,150,3,1) ，则可以进行广播。例如在张量流中：

`input = tf.expand_dims(input , -1)`

【讨论】：