tfp.experimental.joint_distribution_layers.make_lambda_layer_class

Stay organized with collections Save and categorize content based on your preferences.

Creates a LambdaLayer class.

fn A callable.

lambda_layer_class A new LambdaLayer class that closes over fn.