img_input = torch.zeros([6B, 3, 900, 400], dtype=torch.int32, device=f'cuda:{args.gpu_id}')
请问在转onnx时为什么要设置成900400，不直接设置为900*1600？