code: def _test():
      """
      Simple test
      """
      from labml.logger import inspect
  
      x = torch.zeros([2, 6, 2, 4])
      inspect(x.shape)
      bn = GroupNorm(2, 6)
  
      x = bn(x)
      inspect(x.shape)
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/normalization/group_norm/__init__.py
imports: import torch
  from torch import nn
  from labml_helpers.module import Module
label_desc: Use image node_id 1 for calling a global function with example usage: _test() without return types