code: def test_compute_two_layers_statistic(self):
      self._run_n_batch(num_batch=1)
      self._test_batch_stats(self.first_batch_expected_stats)
  
      self._run_n_batch(num_batch=2)
      self._test_batch_stats(self.second_batch_expected_stats)
  
      self._run_n_batch(num_batch=3)
      self._test_batch_stats(self.third_batch_expected_stats)
  
imports: import math
  from tempfile import TemporaryDirectory
  from unittest import TestCase, skipIf
  from unittest.mock import ANY, MagicMock, call
  import torch
  import torch.nn
  from tests.framework.tools import some_data_generator
  from poutyne import GradientTracker, Model, TensorBoardGradientTracker, WeightsGradientsStatsTracker
label_desc: Use image node_id 2 for calling the GradientStatsTrackerTest obj's underlying member method code with example usage: obj.test_compute_two_layers_statistic() without return types