PyTorch torch.kron 函数

PyTorch torch.kron 函数 Pytorch torch 参考手册 torch.kron 是 PyTorch 中用于计算两个张量 Kronecker 积的函数。Kronecker 积会将第一个张量的每个元素与第二个张量相乘并重新排列。函数定义 torch.kron(input, other) 参数说明： input: 第一个输入张量 other: 第二个输入张量使用示例实例 [mycode4..

更新于 2026/7/16

教程详情页默认 noindex。

PyTorch torch.kron 函数

Pytorch torch 参考手册

torch.kron 是 PyTorch 中用于计算两个张量 Kronecker 积的函数。Kronecker 积会将第一个张量的每个元素与第二个张量相乘并重新排列。

函数定义

torch.kron(input, other)

参数说明：

input: 第一个输入张量
other: 第二个输入张量

使用示例

实例

import torch

# 创建两个一维张量
a = torch.tensor([1, 2])
b = torch.tensor([3, 4])

# 计算 Kronecker 积
y = torch.kron(a, b)
print(y)

输出结果为：

tensor([3, 4, 6, 8])

实例

import torch

# 创建两个二维张量
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])

# 计算 Kronecker 积
y = torch.kron(a, b)
print(y)

输出结果为：

tensor([[ 5,  6, 10, 12],
        [ 7,  8, 14, 16],
        [15, 18, 20, 24],
        [21, 24, 28, 32]])

Pytorch torch 参考手册