unchecked_layer_call

unchecked_layer_call#

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

Shape-checking opt-out decorator for Layer.__call__.