Python深度学习一书中: 8.4代码VAE在tensorflow2.0错误‘lambda_1/random_normal/shape‘的解决方案

Python深度学习一书中:
8.4代码VAE在tensorflow2.0下会有Duplicate node name in graph: 'lambda_1/random_normal/shape’问题。
多方求助没有得到答案。后偶在一处博客受到启发,感叹自己对tf2不熟悉的程度,有再空从头理一遍吧。
主要是将原书254页的

z=layers.Lambda(simpling)([z_mean,z_log_var])

修改为

z=layers.Lambda(simpling,output_shape=(latent_dim,))([z_mean,z_log_var])

之后即可正常运行代码。

你可能感兴趣的:(技术)