pytorch生成多维对角阵

可以使用PyTorch的`torch.eye()`函数来生成多维对角阵。下面是一个示例代码：

import torch

# 生成一个4×4的单位矩阵（对角阵）
x = torch.eye(4)
print(x)

运行以上代码，将会输出一个4×4的单位矩阵，其中对角线上的元素都为1，其他元素都为0：

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

这样就生成了一个4×4的多维对角阵。