将 Torch 中的 AAABBB 张量表格转换为嵌套的 ABABAB 张量表格

我一直在尝试解决这个问题，但仍然无法想出解决方案，也许有人可以在这里帮助我。我有以下输入表格传递给神经网络：

{
  1 :
    {
      1 : DoubleTensor - size: 32x200
      2 : DoubleTensor - size: 32x200
      3 : DoubleTensor - size: 32x200
    }
  2 :
    {
      1 : DoubleTensor - size: 32x54
      2 : DoubleTensor - size: 32x54
      3 : DoubleTensor - size: 32x54
    }
}

上述表格得到预处理后，需要转换为嵌套的 ABABAB 输入表格：

{
  1 :
    {
      1 : DoubleTensor - size: 32x200
      2 : DoubleTensor - size: 32x54
    }
  2 :
    {
      1 : DoubleTensor - size: 32x200
      2 : DoubleTensor - size: 32x54
    }
  3 :
    {
      1 : DoubleTensor - size: 32x200
      2 : DoubleTensor - size: 32x54
    }
}

我要如何使用Torch表层将 AAABBB 表格转换为网络内部的 ABABAB 嵌套表格呢？