Squeeze on first dimension when export onnx and convert tensorrt

There is very common usage:

a = a.squeeze(dim=0)

this op export to onnx, can not convert to tensorrt since tensorrt doesn’t support operation on the first dimension.

Does there anyway to make it done?

Do you have this operation inside the model?
If so, would it be possible to move it outside of it and add the batch dimension to your tensors beforehand?