class_docstr: [`TFSuppressTokensAtBeginLogitsProcessor`] suppresses a list of tokens as soon as the `generate` function starts
  generating using `begin_index` tokens. This should ensure that the tokens defined by `begin_suppress_tokens` at not
  sampled at the begining of the generation.
class_docstr_tok: ["[","`","TFSuppressTokensAtBeginLogitsProcessor","`","]","suppresses","a","list","of","tokens","as","soon","as","the","`","generate","`","function","starts","generating","using","`","begin_index","`","tokens",".","This","should","ensure","that","the","tokens","defined","by","`","begin_suppress_tokens","`","at","not","sampled","at","the","begining","of","the","generation","."]
code: def __call__(
      self, input_ids: tf.Tensor, scores: tf.Tensor, cur_len: int
  ) -> tf.Tensor:
      scores = tf.cond(
          tf.equal(cur_len, self.begin_index),
          lambda: tf.tensor_scatter_nd_update(
              scores,
              indices=[
                  [i, token]
                  for i in range(scores.shape[0])
                  for token in self.begin_suppress_tokens
              ],
              updates=[
                  -float("inf")
                  for _ in range(
                      scores.shape[0] * len(self.begin_suppress_tokens)
                  )
              ],
          ),
          lambda: scores,
      )
      return scores
  
imports: import inspect
  from typing import List, Tuple
  import numpy
  import tensorflow
  from .tf_utils import stable_softmax
  from .utils import add_start_docstrings
  from .utils.logging import get_logger
label_desc: Use image node_id 2 for calling the TFSuppressTokensAtBeginLogitsProcessor obj's underlying member method code with example usage: obj.__call__(input_ids, scores, cur_len) and returns: scores