Не удалось построить TypeSpec для матрицы, когда модель прогнозирует

Question

Модель отлично работает на TensorFlow 1.15 (новейшая версия), но не работает на Tensorflow 2.0 при попытке вызвать прогнозирование. Модель представляет собой модель Keras.

model.predict(self.A.todense())

Извлеченное сообщение об ошибке:

/ tenorflow-2.0.0 / python3.6 / tenorflow_core / python / framework / type_spec.py вtype_spec_from_value (значение)

490 
491   raise TypeError("Could not build a TypeSpec for %r with type %s" %

-> 492 (значение, тип (значение). name ))

493 
494

TypeError: Не удалось создать TypeSpecдля матрицы ([[0, 1, 0, ..., 0, 0, 0],

    [0, 0, 0, ..., 0, 0, 0],
    [0, 0, 0, ..., 0, 0, 0],
    ...,
    [0, 0, 0, ..., 0, 0, 0],
    [0, 0, 0, ..., 0, 0, 0],
    [0, 0, 0, ..., 0, 0, 0]], dtype=int64) with type matrix