code: def pack_scales_and_zeros(scales, zeros):
      assert scales.shape == zeros.shape
      assert scales.dtype == torch.bfloat16
      assert zeros.dtype == torch.bfloat16
      return (
          torch.cat(
              [
                  scales.reshape(scales.size(0), scales.size(1), 1),
                  zeros.reshape(zeros.size(0), zeros.size(1), 1),
              ],
              2,
          )
          .transpose(0, 1)
          .contiguous()
      )
  
imports: import time
  from pathlib import Path
  import torch
  import torch.nn
  import torch.nn.functional
  from sentencepiece import SentencePieceProcessor
  from model import Transformer
label_desc: Use image node_id 3 for calling a global function with example usage: pack_scales_and_zeros(scales, zeros) and returns: torch