code: def dropout(input_tensor, dropout_prob):
      """Perform dropout.
  
      Args:
          input_tensor: float Tensor.
          dropout_prob: Python float. The probability of dropping out a value (NOT of
              *keeping* a dimension as in `tf.nn.dropout`).
  
      Returns:
          A version of `input_tensor` with dropout applied.
      """
      if dropout_prob is None or dropout_prob == 0.0:
          return input_tensor
      output = tf.nn.dropout(input_tensor, dropout_prob)
      return output
  
imports: import copy
  import json
  import math
  import numpy
  import tensorflow
  import functools
  from transformers.activations_tf import gelu
  from finetune.optimizers.recompute_grads import recompute_grad
  from finetune.nn.auxiliary import embed_position
label_desc: Use image node_id 3 for calling a global function with example usage: dropout(input_tensor, dropout_prob) and returns: output, input_tensor