torch.kron¶
- torch.kron(input, other, *, out=None) Tensor ¶
计算
input
和other
的克罗内克积,记为 。如果
input
是一个形状为 张量,并且other
是一个形状为 张量,则结果将是一个形状为 张量,其条目如下:其中 对于 。如果一个张量的维度少于另一个,则会对其进行扩展,直到其具有相同的维度数量。
支持实值和复值输入。
注意
此函数将克罗内克积的典型定义(针对两个矩阵)推广到两个张量,如上所述。当
input
是一个 矩阵,并且other
是一个 矩阵时,结果将是一个 块矩阵。其中
input
是 且other
是 .示例
>>> 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.]])