PyTorch torch.kron-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.kronist eine Funktion in PyTorch, die verwendet wird, um das Kronecker-Produkt zweier Tensoren zu berechnen. Das Kronecker-Produkt multipliziert jedes Element des ersten Tensors mit dem zweiten Tensor und ordnet sie neu an.

Funktionsdefinition

torch.kron(input, other)

Parameterbeschreibung:

  • input: Der erste Eingabetensor
  • other: Der zweite Eingabetensor

Anwendungsbeispiele

Beispiel

import torch

# Zwei eindimensionale Tensoren erstellen
a = torch.tensor([1, 2])
b = torch.tensor([3, 4])

# Das Kronecker-Produkt berechnen
y = torch.kron(a, b)
print(y)

Das Ausgabeergebnis ist:

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

Beispiel

import torch

# Zwei zweidimensionale Tensoren erstellen
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])

# Das Kronecker-Produkt berechnen
y = torch.kron(a, b)
print(y)

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen