pmac123
Results
1
comments of
pmac123
after version 0.12 use the "convert_to_tensor_fn" for outputs' manipulation. ex: dist1 = tfp.layers.DistributionLambda( make_distribution_fn=my_dist, convert_to_tensor_fn=tfp.distributions.Distribution.mean )(params) model_mean_sd_1 = Model(inputs=inputs, outputs=dist1)