code: def rot(x):
      return torch.stack([-x[..., 1::2], x[..., ::2]], -1).reshape(
          x.shape
      )
  
imports: import math
  from typing import List, Tuple, Optional, Union
  import torch
  from torch import nn
  from .trace_utils import _assert
label_desc: Use image node_id 5 for calling a global function with example usage: rot(x) and returns: torch