Python PyTorch kron用法及代碼示例

本文簡要介紹python語言中 torch.kron 的用法。

用法:

torch.kron(input, other, *, out=None) → Tensor

參數：

• input(Tensor) -

• other(Tensor) -

關鍵字參數：

out(Tensor,可選的) -輸出張量。如果 None 則忽略。默認值：None

計算 input 和 other 的克羅內克積，用 \otimes 表示。

如果 input 是 (a_0 \times a_1 \times \dots \times a_n) 張量且 other 是 (b_0 \times b_1 \times \dots \times b_n) 張量，則結果將是具有以下條目的 (a_0*b_0 \times a_1*b_1 \times \dots \times a_n*b_n) 張量：

(\text{input} \otimes \text{other})_{k_0, k_1, \dots, k_n} = \text{input}_{i_0, i_1, \dots, i_n} * \text{other}_{j_0, j_1, \dots, j_n},

其中 k_t = i_t * b_t + j_t 為 0 \leq t \leq n 。如果一個張量的維數比另一個少，則它不會被壓縮，直到它具有相同的維數。

支持實值和complex-valued 輸入。

注意

如上所述，此函數將兩個矩陣的 Kronecker 積的典型定義推廣到兩個張量。當input 是(m \times n) 矩陣並且other 是(p \times q) 矩陣時，結果將是(p*m \times q*n) 塊矩陣：

\mathbf{A} \otimes \mathbf{B}=\begin{bmatrix} a_{11} \mathbf{B} & \cdots & a_{1 n} \mathbf{B} \\ \vdots & \ddots & \vdots \\ a_{m 1} \mathbf{B} & \cdots & a_{m n} \mathbf{B} \end{bmatrix}

其中 input 是 \mathbf{A} 和 other 是 \mathbf{B} 。

例子：

>>> mat1 = torch.eye(2)
>>> mat2 = torch.ones(2, 2)
>>> torch.kron(mat1, mat2)
tensor([[1., 1., 0., 0.],
        [1., 1., 0., 0.],
        [0., 0., 1., 1.],
        [0., 0., 1., 1.]])

>>> mat1 = torch.eye(2)
>>> mat2 = torch.arange(1, 5).reshape(2, 2)
>>> torch.kron(mat1, mat2)
tensor([[1., 2., 0., 0.],
        [3., 4., 0., 0.],
        [0., 0., 1., 2.],
        [0., 0., 3., 4.]])