code: def test_swish_takes_float(self) -> None:
      m = SwishLayerNorm([3, 4])
      input = torch.randn(2, 3, 4)
      output = m(input)
      norm = torch.nn.LayerNorm([3, 4])
      ref_output = input * torch.sigmoid(norm(input))
      self.assertTrue(torch.allclose(output, ref_output))
  
imports: import unittest
  import torch
  from torchrec.fx import Tracer
  from torchrec.modules.activation import SwishLayerNorm
label_desc: Use image node_id 1 for calling the TestActivation obj's underlying member method code with example usage: obj.test_swish_takes_float() without return types