numpy.eye()
函数返回一个二维数组(也称为矩阵),其主对角线上的元素为1,其余元素为0
例如,当你调用 numpy.eye(3)
时,将返回以下 3x3 矩阵:
[[1, 0, 0],
[0, 1, 0],
[0, 0, 1]]
请注意,numpy.eye()
函数还可以接受其他参数来创建具有不同特性的单位矩阵。要了解更多关于此函数的信息,请查阅 NumPy 官方文档:https://numpy.org/doc/stable/reference/generated/numpy.eye.html