Функция torch.kron в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.kron— это функция в PyTorch, используемая для вычисления произведения Кронекера двух тензоров. Произведение Кронекера умножает каждый элемент первого тензора на второй тензор и переупорядочивает их.

Определение функции

torch.kron(input, other)

Описание параметров:

  • input: Первый входной тензор
  • other: Второй входной тензор

Примеры использования

Пример

import torch

# Создайте два одномерных тензора
a = torch.tensor([1, 2])
b = torch.tensor([3, 4])

# Вычислить произведение Кронекера
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]])

# Вычислить произведение Кронекера
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 参考手册Справочное руководство Pytorch torch

Другие расширения