(self, input_signature)
Returns newly initialized weights for this layer.

    Weights is a single  `w` tensor with previously specified shape.

    Args:
      input_signature: `ShapeDtype` instance characterizing the input this layer
          should compute on. Unused.
Returns newly initialized weights for this layer.
Returns newly initialized weights for this layer .
def init_weights_and_state(self, input_signature):
    """Returns newly initialized weights for this layer.

    Weights is a single  `w` tensor with previously specified shape.

    Args:
      input_signature: `ShapeDtype` instance characterizing the input this layer
          should compute on. Unused.
    """
    del input_signature  # Unused. There is no input to this layer.
    self.weights = self._initializer(self._shape, self.rng)
    if self._use_bfloat16:
      self.weights = self.weights.astype(jnp.bfloat16)
def init_weights_and_state ( self , input_signature ) : del input_signature # Unused. There is no input to this layer. self . weights = self . _initializer ( self . _shape , self . rng ) if self . _use_bfloat16 : self . weights = self . weights . astype ( jnp . bfloat16 )