code: def _uniform_distribution(
      mean_anchor_probabilities: Tensor,
  ) -> Tensor:
      dim = mean_anchor_probabilities.shape[0]
      return mean_anchor_probabilities.new_ones(dim) / dim
  
file_path: lightly/tests/loss/test_PMSNLoss.py
imports: import math
  import unittest
  import pytest
  import torch
  from torch import Tensor
  from lightly.loss import pmsn_loss
  from lightly.loss.pmsn_loss import PMSNCustomLoss, PMSNLoss
label_desc: Use image node_id 2 for calling a global function with example usage: _uniform_distribution(mean_anchor_probabilities) and returns: unknown