


torch.argmin(a)
      tensor(13)
      >>> torch.argmin(a, dim=1)
      tensor([ 2,  1,  3,  1])
      >>> torch.argmin(a, dim=1, keepdim=True)
      tensor([[2],
              [1],
              [3],
              [1]])


