ValueError : 요소가 두 개 이상있는 배열의 진리 값이 모호합니다. 4D 배열에서 1과 0 카운트를 찾는 방법

나는 예측과 근거가있는 세분화 평가 작업을 계산하고 있습니다. 예측 배열 및 Ground Truth 배열 모양 (28, 1, 224, 224). 이제 참 양성 (TP), 참 음성 (TN), 거짓 양성 (FP), 거짓 음성 (FN)을 계산합니다. 다음 코드를 사용하여 이러한 값을 계산하는 데 사용했지만 오류가 있습니다.

def perf_measure( y_hat , y_actual):
TP = 0
FP = 0
TN = 0
FN = 0


for i in range(len(y_hat)): 
    
    if y_actual[i]==1 and y_hat[i]==1:
       TP += 1
    if y_hat[i]==1 and y_actual[i]!=y_hat[i]:
       FP += 1
    if y_actual[i]==0 and y_hat[i]==0:
       TN += 1
    if y_hat[i]==0 and y_actual[i]!=y_hat[i]:
       FN += 1

return(TP, FP, TN, FN)