code: def argmax_over_dim_0(x: Tensor) -> Tensor:
      return torch.argmax(x, dim=0)
  
imports: import collections
  import dataclasses
  import functools
  import warnings
  import logging
  from typing import Optional, Sequence, Union, Dict, List, Any, Iterable, Callable
  import numpy
  import torch
  from torch import nn, Tensor
  from torch.utils.data import Dataset, ConcatDataset
  from torch.utils.data.dataloader import default_collate
  from .support import pytorch_toolbelt_deprecated
label_desc: Use image node_id 5 for calling a global function with example usage: argmax_over_dim_0(x) and returns: torch