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