code: def test_forward_cuda(self) -> None:
      torch.manual_seed(0)
      criterion = PMSNLoss()
      anchors = torch.rand((8 * 3, 10)).cuda()
      targets = torch.rand((8, 10)).cuda()
      prototypes = torch.rand((4, 10), requires_grad=True).cuda()
      criterion(anchors, targets, prototypes)
  
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 3 for calling the TestPMSNLoss obj's underlying member method code with example usage: obj.test_forward_cuda() without return types