【寒武纪硬件产品型号】必填*：MLU370-S4

【使用操作系统】必填*：麒麟ky10(centos8)

【使用驱动版本】必填*：v.6.2.10

【出错信息】必填*：encoding和decoding慢2S左右

【当前已做了哪些信息确认】选填：

1. GPU（3090TI）运行速度统计

2. 使用方式，使用torch2mlu脚本转换sam代码然后因为有报错，数据调整成float32，

===================================================

def batched_mask_to_box(masks: torch.Tensor) -> torch.Tensor:

    """

    Calculates boxes in XYXY format around masks. Return [0,0,0,0] for

    an empty mask. For input shape C1xC2x...xHxW, the output shape is C1xC2x...x4.

    """

    if torch.numel(masks) == 0:

        return torch.zeros(*masks.shape[:-2], 4, device=masks.device)

    masks = masks.to(torch.float32) # 额外添加该行，才能执行。

==================================================