(self, input_signature)
Initializes this layer's weights.
Initializes this layer's weights.
Initializes this layer s weights .
def init_weights_and_state(self, input_signature):
    """Initializes this layer's weights."""
    if isinstance(input_signature, (list, tuple)):
      input_signature = input_signature[0]
    self.weights = ()
    self.state = (jnp.zeros(input_signature.shape, dtype=jnp.int32),
                  {'running_second_time': ()})
def init_weights_and_state ( self , input_signature ) : if isinstance ( input_signature , ( list , tuple ) ) : input_signature = input_signature [ 0 ] self . weights = ( ) self . state = ( jnp . zeros ( input_signature . shape , dtype = jnp . int32 ) , { 'running_second_time' : ( ) } )