Вам следует проверить фактическое значение вашего tuple
-типа x_train.shape
. Простая печать, вероятно, приведет к тому, что ваш x_train.shape
имеет форму (N,)
, то есть имеет одноэлементное измерение.
В случае, если это ваша проблема, вам нужно expand_dims ваш (вероятно, ) numpy массив.