code: def swap_axes(
      x: ep.TensorType, dim0: int, dim1: int
  ) -> ep.TensorType:
      assert dim0 < x.ndim
      assert dim1 < x.ndim
  
      axes = list(range(x.ndim))
      axes[dim0] = dim1
      axes[dim1] = dim0
  
      return ep.transpose(x, tuple(axes))
  
imports: from typing import Union, List, Tuple
  import eagerpy
label_desc: Use image node_id 5 for calling a global function with example usage: swap_axes(x, dim0, dim1) and returns: ep