f1_score(label,prediction)传入的两个参数都是一维的tensor,所以如果是多维的tensor,需要先将多维的展平(调用flatten()方法)为一维的tensor
注意:flatten()函数要求的参数类型是Tensor,不能是numpy里的array数组
所以如果拿到的数据是array数组,需要先将array数组转化成tensor
TypeError: flatten(): argument 'input' (position 1) must be Tensor, not numpy.ndarray
label_tensor=torch.from_numpy(label_array)
prediction_tensor=torch.from_numpy(prediction_array)
label_final=torch.flatten(label_tensor)
prediction_final=torch.flatten(prediction_tensor)
fl_score(label_final,prediction_final)