Ekaterina_Dranitsyna:

tf.keras.utils.plot_model(model, show_shapes=True, show_dtype=True)


Thanks, that could be possibility but I really want to print the dataset dimension just like it in pytorch. Provided if it is possible at all?