code: def get_labels(inners: jnp.ndarray) -> Tuple[np.ndarray, np.ndarray]:
      """Labels the inner products compute dfrom `compute_inners`."""
      bs = inners.shape[0]
      labels = np.zeros(
          inners.reshape((bs, bs, -1)).shape, dtype=inners.dtype
      )
      labels_all = np.zeros_like(labels)
      for i in range(inners.shape[0]):
          labels[i, i, 0] = 1
          labels_all[i, i, :] = 1
      return labels.reshape(inners.shape), labels_all.reshape(
          inners.shape
      )
  
file_path: scenic/scenic/projects/verbs_in_action/losses.py
imports: from typing import Tuple
  import flax.linen
  import jax.numpy
  import numpy
  from scenic.projects.verbs_in_action import utils
label_desc: Use image node_id 8 for calling a global function with example usage: get_labels(inners) and returns: labels,  labels_all