code: def _weights(self) -> torch.Tensor:
      return nn.functional.softmax(self.weight_logits, dim=0)
  
imports: from torch import nn
  import torch
label_desc: Use image node_id 3 for calling the GMMModel obj's underlying member method code with example usage: obj._weights() and returns: nn