code: def test_shape_assertion_forward(self) -> None:
      bs = 3
      dim = 128
      k = 32
  
      loss_fn = MMCRLoss()
      online = torch.randn(bs, k, dim)
      momentum = torch.randn(bs, k, dim + 1)
  
      with self.assertRaises(AssertionError):
          loss_fn(online, momentum)
  
file_path: lightly/tests/loss/test_MMCR_loss.py
imports: import unittest
  import torch
  from lightly.loss.mmcr_loss import MMCRLoss
label_desc: Use image node_id 5 for calling the testMMCRLoss obj's underlying member method code with example usage: obj.test_shape_assertion_forward() without return types