PyTorch torch.diag_embed Funktion
PyTorch torch Referenzhandbuch
torch.diag_embedist eine Funktion in PyTorch, die verwendet wird, um den Eingabe-Tensor als diagonale Elemente in einen neuen Tensor einzubetten. Sie erstellt eine Diagonale in der letzten Dimension der Eingabe.
Funktionsdefinition
torch.diag_embed(input, diagonal=0, offset=0)
Parameterbeschreibung:
input: Eingabetensordiagonal/offset: Diagonalindex
Anwendungsbeispiele
Beispiel
import torch
# Einen eindimensionalen Tensor erstellen
x = torch.tensor([1, 2, 3])
# Diagonaleinbettung erstellen
y = torch.diag_embed(x)
print(y)
# Einen eindimensionalen Tensor erstellen
x = torch.tensor([1, 2, 3])
# Diagonaleinbettung erstellen
y = torch.diag_embed(x)
print(y)
Die Ausgabe ist:
tensor([[1, 0, 0],
[0, 2, 0],
[0, 0, 3]])
Beispiel
import torch
# Einen zweidimensionalen Tensor erstellen
x = torch.tensor([[1, 2], [3, 4]])
# Diagonaleinbettung erstellen
y = torch.diag_embed(x)
print(y.shape)
print(y)
# Einen zweidimensionalen Tensor erstellen
x = torch.tensor([[1, 2], [3, 4]])
# Diagonaleinbettung erstellen
y = torch.diag_embed(x)
print(y.shape)
print(y)
Die Ausgabe ist:
torch.Size([2, 2, 2])
tensor([[[1, 0],
[0, 2]],
[[3, 0],
[0, 4]]])
Andere Erweiterungen