checked_layer_call

checked_layer_call#

penzai.core.layer.checked_layer_call(func: Callable[[Layer, Any], Any]) Callable[[Layer, Any], Any][source]#

Decorator for Layer.__call__ to add shape-checking and name scopes.