ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

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.