numpy.diag 函數在不同版本的 NumPy 中可能存在一些兼容性問題
NumPy 1.7 及更早版本：在這些版本中，numpy.diag 函數只接受一個參數，即要提取對角線元素的矩陣。如果你需要從多維數組中提取對角線元素，請使用 numpy.diagonal 函數。
NumPy 1.8 及更高版本：在這些版本中，numpy.diag 函數已經得到了改進，現在可以接受兩個參數。第一個參數是要提取對角線元素的矩陣，第二個參數是可選的，表示要提取的對角線的偏移量。默認值為 0，表示主對角線。正值表示上對角線，負值表示下對角線。
為了確保代碼的兼容性，你可以根據 NumPy 的版本來選擇合適的函數。例如：
import numpy as np
if np.__version__ < '1.8':
# 使用 numpy.diagonal 函數
diagonal_elements = np.diagonal(matrix)
else:
# 使用 numpy.diag 函數
diagonal_elements = np.diag(matrix)
或者，你可以直接使用 numpy.diag 函數，并在需要時提供第二個參數來指定對角線的偏移量。這樣，你的代碼將在 NumPy 1.7 及更高版本中都能正常工作。
