tfp.experimental.joint_distribution_layers.make_lambda_layer_class

Creates a LambdaLayer class.

fn A callable.

lambda_layer_class A new LambdaLayer class that closes over fn.