code: def test_gstg_1d_input_with_reg_reduction(self) -> None:
      dim = 3
      mean_gstg = GaussianStochasticGates(dim, reg_reduction="mean").to(
          self.testing_device
      )
      none_gstg = GaussianStochasticGates(dim, reg_reduction="none").to(
          self.testing_device
      )
  
      input_tensor = torch.tensor(
          [
              [0.0, 0.1, 0.2],
              [0.3, 0.4, 0.5],
          ]
      ).to(self.testing_device)
  
      _, mean_reg = mean_gstg(input_tensor)
      _, none_reg = none_gstg(input_tensor)
      expected_mean_reg = 0.8404
      expected_none_reg = torch.tensor([0.8424, 0.8384, 0.8438])
  
      assertTensorAlmostEqual(self, mean_reg, expected_mean_reg)
      assertTensorAlmostEqual(self, none_reg, expected_none_reg)
  
imports: import unittest
  import torch
  from captum.module.gaussian_stochastic_gates import GaussianStochasticGates
  from parameterized import parameterized_class
  from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
label_desc: Use image node_id 3 for calling the TestGaussianStochasticGates obj's underlying member method code with example usage: obj.test_gstg_1d_input_with_reg_reduction() without return types