constructor({ logits, pred_boxes, pred_masks }) {
        super();
        this.logits = logits;
        this.pred_boxes = pred_boxes;
        this.pred_masks = pred_masks;
    }
@param {Object} output The output of the model.
@param {Tensor} output.logits The output logits of the model.
@param {Tensor} output.pred_boxes Predicted boxes.
@param {Tensor} output.pred_masks Predicted masks.