parameters: (self)
docstring: Size of the last dimension of the logits `Tensor`.
  
      Typically, logits is of shape `[batch_size, logits_dimension]`.
  
      Returns:
        The expected size of the `logits` tensor.
docstring_summary: Size of the last dimension of the logits `Tensor`.
docstring_tokens: Size of the last dimension of the logits Tensor .
function: def logits_dimension(self):
      """Size of the last dimension of the logits `Tensor`.
  
      Typically, logits is of shape `[batch_size, logits_dimension]`.
  
      Returns:
        The expected size of the `logits` tensor.
      """
      raise NotImplementedError("Calling an abstract method.")
function_tokens: def logits_dimension ( self ) : raise NotImplementedError ( "Calling an abstract method." )