code: def test_get_gate_active_probs_2d_input_with_mask(self) -> None:
      dim = 3
      mask = torch.tensor(
          [
              [0, 1],
              [1, 1],
              [0, 2],
          ]
      )
      bcstg = BinaryConcreteStochasticGates(dim, mask=mask).to(
          self.testing_device
      )
  
      input_tensor = torch.tensor(
          [
              [
                  [0.0, 0.1],
                  [0.2, 0.3],
                  [0.4, 0.5],
              ],
              [
                  [0.6, 0.7],
                  [0.8, 0.9],
                  [1.0, 1.1],
              ],
          ]
      ).to(self.testing_device)
  
      bcstg(input_tensor)
      gate_active_probs = bcstg.get_gate_active_probs()
  
      expected_gate_active_probs = [0.8319, 0.8324, 0.8304]
  
      assertTensorAlmostEqual(
          self,
          gate_active_probs,
          expected_gate_active_probs,
          mode="max",
      )
  
imports: import unittest
  import torch
  from captum.module.binary_concrete_stochastic_gates import BinaryConcreteStochasticGates
  from parameterized import parameterized_class
  from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
label_desc: Use image node_id 19 for calling the TestBinaryConcreteStochasticGates obj's underlying member method code with example usage: obj.test_get_gate_active_probs_2d_input_with_mask() without return types