code: def forward(self, x):
      wg = self.wg.float() if self.fp32_gate else self.wg
      return wg(x.to(dtype=wg.weight.dtype))
  
imports: import torch
label_desc: Use image node_id 2 for calling the LinearTopKGate obj's underlying member method code with example usage: obj.forward(x) and returns: wg