code: def lower_triangular_matrix(diag, non_diag):
      assert isinstance(diag, torch.Tensor)
      assert isinstance(non_diag, torch.Tensor)
      batch_size = diag.shape[0]
      n = diag.shape[1]
      y = torch.zeros((batch_size, n, n), dtype=torch.float32)
      y = y.to(diag.device)
      set_batch_non_diagonal(y, non_diag)
      set_batch_diagonal(y, diag)
      return y
  
imports: import numpy
  import torch
label_desc: Use image node_id 3 for calling a global function with example usage: lower_triangular_matrix(diag, non_diag) and returns: y