Pytorch Torch Diag

📅 2026-06-22 | 📂 PyTorch

# Create 1D Tensor x = torch.tensor([1,2,3]) # Create Diagonal Matrix y = torch.diag(x) print(y)

Output result is:

tensor([[1, 0, 0],
        [0, 2, 0],
        [0, 0, 3]])

Example

import torch

# Extract Diagonal Elements from Matrix

 x = torch.tensor([[1,2,3],[4,5,6],[7,8,9]])

# Extract Main Diagonal

 y = torch.diag(x)

print(y)

Output result is:

tensor([1, 5, 9])

Image 4: Pytorch torch Reference Manual Pytorch torch Reference Manual

YouTip

Pytorch Torch Diag

Example

📂 Categories