code: def invariance_loss(x: Tensor, y: Tensor) -> Tensor:
      """Returns VICReg invariance loss.
  
      Args:
          x:
              Tensor with shape (batch_size, ..., dim).
          y:
              Tensor with shape (batch_size, ..., dim).
      """
      return F.mse_loss(x, y)
  
file_path: lightly/lightly/loss/vicreg_loss.py
imports: import torch
  import torch.distributed
  import torch.nn.functional
  from torch import Tensor
  from lightly.utils.dist import gather
label_desc: Use image node_id 1 for calling a global function with example usage: invariance_loss(x, y) and returns: F