class_docstr: Dropout with two paths.
code: def __call__(
      self, x, deterministic: bool, use_aux_dropout: bool
  ) -> jnp.ndarray:
      x1 = nn.Dropout(rate=self.rate)(x, deterministic)
      x2 = nn.Dropout(rate=self.aux_rate)(x, deterministic)
      if use_aux_dropout:
          return x2
      else:
          return x1
  
imports: from typing import Any, Callable, Optional, Sequence
  from absl import logging
  import flax
  import flax.linen
  import jax
  import jax.numpy
  import ml_collections
  import numpy
  from scenic.model_lib.base_models.multilabel_classification_model import MultiLabelClassificationModel
  from scenic.model_lib.layers import attention_layers
  from scenic.model_lib.layers import nn_layers
  import scipy
label_desc: Use image node_id 1 for calling the SplitDropout obj's underlying member method code with example usage: obj.__call__(x, deterministic, use_aux_dropout) and returns: x2, x1