(self, input_signature)
Sets layer-specific internal state.
Sets layer-specific internal state.
Sets layer - specific internal state .
def init_weights_and_state(self, input_signature):
    """Sets layer-specific internal state."""
    del input_signature
    self.state = jnp.array(0, dtype=jnp.int32)
def init_weights_and_state ( self , input_signature ) : del input_signature self . state = jnp . array ( 0 , dtype = jnp . int32 )