Керас подкласса Tensorflow с несколькими выходными формами ⇐ Python

[Anonymous]

Код: Выделить всё

class MyModel(tf.keras.Model):
def __init__(self):
super().__init__()
self.dense = tf.keras.layers.Dense(1)
self.build(input_shape=[None, 1])

def call(self, inputs, **kwargs):
return self.dense(inputs)

MyModel().summary()

График модели также не работает:

Код: Выделить всё

tf.keras.utils.plot_model(model, to_file='model_1.png', show_shapes=True)

Я пробовал этот код на нескольких версиях тензорного потока 2.3.0, 2.3.1 и 2.4.1, и каждый раз выходная форма была многократной! Это ошибка? Есть исправления?

Подробнее здесь: https://stackoverflow.com/questions/669 ... tput-shape