code: def accuracy(output: torch.Tensor, labels: torch.Tensor):
      """
      A simple function to calculate the accuracy
      """
      return output.argmax(dim=-1).eq(labels).sum().item() / len(labels)
  
imports: from typing import Dict
  import numpy
  import torch
  from torch import nn
  from labml import lab, monit, tracker, experiment
  from labml.configs import BaseConfigs, option, calculate
  from labml.utils import download
  from labml_helpers.device import DeviceConfigs
  from labml_helpers.module import Module
  from labml_nn.graphs.gat import GraphAttentionLayer
  from labml_nn.optimizers.configs import OptimizerConfigs
label_desc: Use image node_id 1 for calling a global function with example usage: accuracy(output, labels) and returns: unknown