预测结果是一个多维数组,必须要打平才能计算loss吗?

问题遇到的现象和发生背景

在深度学习项目中,测试集预测结果是一个多维数组,必须要打平才能计算loss和评估吗?

数组如下:

img