code: def test_forward(self) -> None:
      bs = 512
      dim = 128
      num_samples = 2
  
      loss_fn = WMSELoss()
      x = torch.randn(bs * num_samples, dim)
  
      loss_fn(x)
  
file_path: lightly/tests/loss/test_WMSELoss.py
imports: import unittest
  import torch
  from lightly.loss.wmse_loss import WMSELoss
label_desc: Use image node_id 1 for calling the testWMSELoss obj's underlying member method code with example usage: obj.test_forward() without return types