code: def _split_channels(total_filters, num_groups):
      """
      https://github.com/tensorflow/tpu/blob/master/models/official/mnasnet/mixnet/custom_layers.py#L33
      """
      split = [total_filters // num_groups for _ in range(num_groups)]
      split[0] += total_filters - sum(split)
      return split
  
imports: import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 for calling a global function with example usage: _split_channels(total_filters, num_groups) and returns: split