如何将对角矩阵转换为4模张量

Question

我想创建一个函数，将三对角矩阵作为参数并将其转换为4模Tensor，并且Tensor的大小不得大于origianl矩阵。