code: def test_av_load_all_layers_one_identifier(self) -> None:
      with tempfile.TemporaryDirectory() as tmpdir:
          av_01 = torch.randn(36, 16)
          av_02 = torch.randn(16, 16)
          av_03 = torch.randn(4, 16)
          avs_0 = [av_01, av_02, av_03]
  
          av_11 = torch.randn(36, 16)
          av_12 = torch.randn(16, 16)
          av_13 = torch.randn(4, 16)
          avs_1 = [av_11, av_12, av_13]
  
          idf1, idf2 = "idf1", "idf2"
  
          AV.save(
              tmpdir,
              "dummy",
              idf1,
              ["layer1.0.conv1", "layer1.0.conv2", "layer1.1.conv1"],
              avs_0,
              "0",
          )
          dataloader = DataLoader(
              cast(Dataset, AV.load(tmpdir, "dummy"))
          )
          self.assertEqual(len(dataloader), 3)
  
          AV.save(
              tmpdir,
              "dummy",
              idf2,
              ["layer1.0.conv1", "layer1.0.conv2", "layer1.1.conv1"],
              avs_1,
              "0",
          )
          dataloader = DataLoader(
              cast(Dataset, AV.load(tmpdir, "dummy"))
          )
          self.assertEqual(len(dataloader), 6)
  
          # check activations for idf1
          dataloader_layer = DataLoader(
              cast(Dataset, AV.load(tmpdir, "dummy", identifier=idf1))
          )
          self.assertEqual(len(dataloader_layer), 3)
  
          for i, av in enumerate(dataloader_layer):
              assertTensorAlmostEqual(self, av, avs_0[i].unsqueeze(0))
  
          # check activations for idf2
          dataloader_layer = DataLoader(
              cast(Dataset, AV.load(tmpdir, "dummy", identifier=idf2))
          )
          self.assertEqual(len(dataloader_layer), 3)
          for i, av in enumerate(dataloader_layer):
              assertTensorAlmostEqual(self, av, avs_1[i].unsqueeze(0))
  
imports: import glob
  import tempfile
  from datetime import datetime
  from typing import cast, List
  import torch
  from captum._utils.av import AV
  from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
  from tests.helpers.basic_models import BasicLinearReLULinear
  from torch.utils.data import DataLoader, Dataset
label_desc: Use image node_id 10 for calling the Test obj's underlying member method code with example usage: obj.test_av_load_all_layers_one_identifier() without return types