def get_mask_invalid(matrix): mask = np.isposinf(matrix) + np.isneginf(matrix) + np.isnan(matrix) return mask