View source on GitHub |
Creates a LambdaLayer
class.
tfp.experimental.joint_distribution_layers.make_lambda_layer_class(
fn
)
Args | |
---|---|
fn
|
A callable. |
Returns | |
---|---|
lambda_layer_class
|
A new LambdaLayer class that closes over fn .
|