def test_loss_value(self) -> None:
        """If all values are zero, the loss should be zero."""
        bs = 3
        dim = 128
        k = 32

        loss_fn = MMCRLoss()
        online = torch.zeros(bs, k, dim)
        momentum = torch.zeros(bs, k, dim)

        loss = loss_fn(online, momentum)
        assert loss == 0.0
If all values are zero, the loss should be zero.
tests/loss/test_MMCR_loss.py