Tensorrt Python Example For Loop