class_docstr: Contrastive Model, currently mainly focused on memory bank and CSC.
class_docstr_tok: ["Contrastive","Model",",","currently","mainly","focused","on","memory","bank","and","CSC","."]
code: def KLDivLoss(self, out, code):
      softmax = nn.Softmax(dim=1).cuda()
      p = softmax(out / self.T)
      loss = torch.mean(-torch.sum(code * torch.log(p), dim=1))
      return loss
  
imports: import math
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  import slowfast.models.losses
  import slowfast.utils.distributed
  import slowfast.utils.logging
  from slowfast.models.video_model_builder import X3D, MViT, ResNet, SlowFast
  from .build import MODEL_REGISTRY
label_desc: Use image node_id 20 for calling the ContrastiveModel obj's underlying member method code with example usage: obj.KLDivLoss(out, code) and returns: loss