code: def make_weight_cp(t, wa, wb):
      temp = torch.einsum("i j k l, j r -> i r k l", t, wb)
      return torch.einsum("i j k l, i r -> r j k l", temp, wa)
  
imports: import torch
label_desc: Use image node_id 1 for calling a global function with example usage: make_weight_cp(t, wa, wb) and returns: torch