code: def get_rotation_x(angle):
      bs = angle.size(0)
      sin, cos = torch.sin(angle), torch.cos(angle)
      mat = torch.zeros((bs * 3, 3)).type_as(sin)
      mat[0:bs, 0] = 1.0
      mat[bs : bs * 2, 1:2], mat[bs : bs * 2, 2:3] = cos, -sin
      mat[bs * 2 : bs * 3, 1:2], mat[bs * 2 : bs * 3, 2:3] = sin, cos
      mat = mat.view(3, bs, 3).permute(1, 0, 2)
      return mat
  
imports: import os
  import torch
  import torch.optim
  import numpy
  from sklearn.metrics import auc
label_desc: Use image node_id 5 for calling a global function with example usage: get_rotation_x(angle) and returns: mat