多batch模型编译

好的大佬,目前我是按照了batch推理这里,输入为2 * 3 * 640 * 640,然后得到了

n_dim: 4 2 80 80 80
n_dim: 4 2 80 80 64
n_dim: 4 2 40 40 80
n_dim: 4 2 40 40 64
n_dim: 4 2 20 20 80
n_dim: 4 2 20 20 64

请问大佬,这里batch推理yolo的输出该怎么解析呢,推理框架是否支持拆分成2个1 * 80 * 80 * 80然后放到decode_box解析呢


如果不支持的话是我只能自己再试下了,现在也不知道batch推理的结果