如何在 Pytorch 中對角地將幾個矩陣組合成一個大矩陣

Question

我有幾個矩陣，比如說[M1,M2,M3,M4] 。 每個矩陣都有不同的形狀。 如何將這些矩陣組合成一個對角線的大矩陣，例如：

[[M1, 0, 0, 0]
 [0, M2, 0, 0]
 [0, 0, M2, 0]
 [0, 0, 0, M2]]

---

例子：

M1 = [[1,2],[2,1]]
M2 = [[1,2]]
M3 = [[3]]
M4 = [[3,4,5],[4,5,6]]

組成這個大矩陣：

[[1, 2, 0, 0, 0, 0, 0]
 [2, 1, 0, 0, 0, 0, 0]
 [0, 0, 1, 2, 0, 0, 0]
 [0, 0, 0, 0, 3, 4, 5]
 [0, 0, 0, 0, 4, 5, 6]]

Answer 1

以下是使用 SciPy 的方法：

from scipy.sparse import block_diag

block_diag((M1, M2, M3, M4))

Answer 2

使用 PyTorch 的torch.block_diag() ：

>>> torch.block_diag(M1,M2,M3,M4)

tensor([[1, 2, 0, 0, 0, 0, 0, 0],
        [2, 1, 0, 0, 0, 0, 0, 0],
        [0, 0, 1, 2, 0, 0, 0, 0],
        [0, 0, 0, 0, 3, 0, 0, 0],
        [0, 0, 0, 0, 0, 3, 4, 5],
        [0, 0, 0, 0, 0, 4, 5, 6]])