torch.diag(x)

torch.diag(x)是一个torch库中的函数，用于创建一个以x为对角线元素的方阵。如果x是一个向量，则返回一个以x为对角线元素的方阵；如果x是一个矩阵，则返回x的对角线元素。以下是一个示例：

import torch

x = torch.tensor([1, 2, 3])  # 创建一个向量
diag_matrix = torch.diag(x)  # 创建一个以x为对角线元素的方阵
print(diag_matrix)

# 输出:
# tensor([[1, 0, 0],
#         [0, 2, 0],
#         [0, 0, 3]])

希望这个能帮到你！如果还有其他问题，请随时提问。