code: def __init__(self, in_channels=4, out_channels=64, kernel_size=3):
      super().__init__()
      self.conv_weight = nn.Parameter(
          torch.randn(
              out_channels, in_channels, kernel_size, kernel_size
          )
      )
  
file_path: ColossalAI/tests/test_auto_parallel/test_tensor_shard/test_metainfo/test_conv_metainfo.py
imports: import pytest
  import torch
  import torch.nn
  from colossalai.device.device_mesh import DeviceMesh
  from colossalai.initialize import launch
  from colossalai.logging import disable_existing_loggers
  from colossalai.testing.pytest_wrapper import run_on_environment_flag
  from colossalai.testing.utils import rerun_if_address_is_in_use, spawn
  from tests.test_auto_parallel.test_tensor_shard.test_metainfo.utils import mem_test_for_node_strategy
label_desc: Use image node_id 1 to create a new ConvFunctionModule object from inherited base classes: nn with example: obj = ConvFunctionModule(in_channels, out_channels, kernel_size)