code: def _bert_add_special_tokens_attention_mask(
      attention_mask, seq_end_col
  ):
      """
      Mark attention mask for special tokens ([CLS] and [SEP]) with 1
      """
      # Copy attention masks for all but last two
      attention_mask[:, 1:-1] = attention_mask[:, 0:-2]
      # Mark [CLS] token with 1
      attention_mask[:, 0] = 1
      # Mark [SEP] token with 1
      attention_mask[
          cp.arange(0, attention_mask.shape[0], dtype=cp.uint32),
          seq_end_col,
      ] = 1
  
file_path: cudf/python/cudf/cudf/core/subword_tokenizer.py
imports: from __future__ import annotations
  import warnings
  from typing import Union
  import cupy
  from cudf._lib.nvtext.subword_tokenize import Hashed_Vocabulary, subword_tokenize_inmem_hash
label_desc: Use image node_id 4 for calling a global function with example usage: _bert_add_special_tokens_attention_mask(attention_mask, seq_end_col) without return types