PyTorch torch.eye-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.eyeist eine Funktion in PyTorch, die zum Erstellen einer Einheitsmatrix verwendet wird (einer Matrix mit 1s auf der Diagonalen und 0s an anderer Stelle).

Dies wird im Deep Learning häufig zur Initialisierung, zum Erstellen von Masken und in anderen Szenarien verwendet.

Funktionsdefinition

torch.eye(n, m, dtype, device, requires_grad)

Parameters:

  • n(int): Anzahl der Zeilen.
  • m(int, optional): Anzahl der Spalten. Wenn nicht angegeben, wird eine quadratische n×n-Matrix erstellt.
  • dtype(torch.dtype, optional): Datentyp.
  • device(torch.device, optional): Gerät.
  • requires_grad(bool, optional): ob die Gradientenberechnung erforderlich ist.

Rückgabewert:

  • torch.Tensor: gibt die Einheitsmatrix zurück.

Anwendungsbeispiele

Beispiel 1: Eine quadratische Matrix erstellen

Beispiel

import torch

# Erstelle eine 3x3-Einheitsmatrix
I = torch.eye(3)

print(I)

Die Ausgabe ist:

tensor([[1., 0., 0.],
        [0., 1., 0.],
        [0., 0., 1.]])

Beispiel 2: Eine nicht-quadratische Matrix erstellen

Beispiel

import torch

# Erstelle eine 3x4-Einheitsmatrix
I = torch.eye(3, 4)

print(I)

Die Ausgabe ist:

tensor([[1., 0., 0., 0.],
        [0., 1., 0., 0.],
        [0., 0., 1., 0.]])

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Weitere Erweiterungen