怎么检查输入矩阵中的NaN以及Inf

可以使用NumPy中的isnan和isinf函数来检查输入矩阵中的NaN和Inf。例如，使用以下代码进行检查:

import numpy as np

a = np.array([[1, 2, np.nan], [4, np.inf, 6]])

print(np.isnan(a))

print(np.isinf(a))

输出结果为：

[[False False True]
[False True False]]

[[False False False]
[False True False]]