reg_max * 4, self.nc), 1) if self.export and self.format == 'edgetpu': # FlexSplitV ops issue x_cat = torch.cat([xi.view(shape[0], self.no, -1) for xi in x], 2) box = x_cat[:, :self.reg_max * 4] cls = x_cat[:, self.reg_max * 4:] else: box, cls = ...