code: def set_conv_prune_dim(dim):
      """
      Parameters:
      dim: int
          0: filter pruning
          1: channel pruning
      """
      global conv_prune_dim
      conv_prune_dim = dim
  
imports: import logging
  import torch
label_desc: Use image node_id 1 for calling a global function with example usage: set_conv_prune_dim(dim) without return types