Fonction torch.kron de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.kronest une fonction dans PyTorch utilisée pour calculer le produit de Kronecker de deux tenseurs. Le produit de Kronecker multiplie chaque élément du premier tenseur par le second tenseur et les réorganise.

Définition de fonction

torch.kron(input, other)

Description des paramètres :

  • input: Le premier tenseur d'entrée
  • other: Le deuxième tenseur d'entrée

Exemples d'utilisation

Exemple

import torch

# Créer deux tenseurs unidimensionnels
a = torch.tensor([1, 2])
b = torch.tensor([3, 4])

# Calculer le produit de Kronecker
y = torch.kron(a, b)
print(y)

Le résultat de sortie est:

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

Exemple

import torch

# Créer deux tenseurs bidimensionnels
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])

# Calculer le produit de Kronecker
y = torch.kron(a, b)
print(y)

Le résultat de sortie est:

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

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions